hdu 1166 敌兵布阵 线段树（单点更新）

链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

题意：中文题，不多说了。

思路：这题以前用树状数组做过，这次用的线段树，当做可参考模板。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<ctype.h>
#include<algorithm>
#include<string>
#define PI acos(-1.0)
#define maxn 100000
#define INF 1<<25
#define MAX 0x7fffffff
typedef long long ll;
using namespace std;
int n,m,l,r;
struct line
{
int left,right;
int n;
int value;
} a [maxn*4];

void build_tree(int l,int r,int step)
{
a[step].left=l;
a[step].right=r;
a[step].n=0;
if(l==r)
{
scanf("%d",&a[step].value);
return;
}
int mid=(l+r)/2;
build_tree(l,mid,step*2);
build_tree(mid+1,r,step*2+1);
a[step].value=a[step*2].value+a[step*2+1].value;
}
int ans;
void add(int s,int step,int k)
{
if(s==a[step].left&&s==a[step].right)
{
a[step].value+=k;
return;
}
if(a[step].left==a[step].right) return;
int mid=(a[step].left+a[step].right)/2;
if(mid>=s) add(s,step*2,k);
else if(mid<s) add(s,step*2+1,k);
else
{
add(s,step*2,k);
add(s,step*2+1,k);
}
a[step].value=a[step*2].value+a[step*2+1].value;
}
int query(int s,int t,int step)
{
if(a[step].left>=s&&a[step].right<=t)
{
ans+=a[step].value;
return 0;
}
else
{
int mid=(a[step].left+a[step].right)/2;
if(mid>=t) query(s,t,step*2);
else if(mid<s) query(s,t,step*2+1);
else
{
query(s,t,step*2);
query(s,t,step*2+1);
}
}
}
int main()
{
int tot,tt;
char ss[10];
scanf("%d",&tot);
for(int ii=1; ii<=tot; ii++)
{
memset(a,0,sizeof(a));
scanf("%d",&tt);
build_tree(1,tt,1);
printf("Case %d:\n",ii);
while(scanf("%s",ss)&&ss[0]!='E')
{
int i,j;
scanf("%d%d",&i,&j);
if(ss[0]=='A')
{
add(i,1,j);
}
else if(ss[0]=='S')
{
add(i,1,-j);
}
else if(ss[0]=='Q')
{
ans=0;
query(i,j,1);
printf("%d\n",ans);
}
}
}
}