hdu 1166 敌兵布阵 (线段树)

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

最基础的线段树，单点更新。完全跟着HH的代码风格写的。

code:

#include<cstdio>
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int maxn = 50005 ;
int sum[maxn<<2] ;
void PushUp(int rt){
sum[rt] = sum[rt<<1] + sum[rt<<1|1] ;
}
void build(int l, int r, int rt){
if(l==r){
scanf("%d", &sum[rt]) ;
return ;
}
int m = (l + r) >> 1 ;
build(lson) ;
build(rson) ;
PushUp(rt) ;
}
void update(int p, int add, int l, int r, int rt){
if(l==r){
sum[rt] += add ;
return ;
}
int m = (l + r) >> 1 ;
if(p<=m) update(p, add, lson) ;
else update(p, add, rson) ;
PushUp(rt) ;
}
int query(int L, int R, int l, int r, int rt){
if(L<=l&&r<=R){
return sum[rt] ;
}
int m = (l + r) >> 1 ;
int ret = 0 ;
if(L<=m) ret += query(L, R, lson) ;
if(R>m) ret += query(L, R, rson) ;
return ret ;
}
int main(){
int t=1, T, n, i, j, a, b ;
char str[10] ;
scanf("%d", &T) ;
while(T--){
scanf("%d", &n) ;
build(1, n, 1) ;
printf("Case %d:\n", t++) ;
while(1){
scanf("%s", str) ;
if(str[0]=='E') break ;
scanf("%d%d", &a, &b) ;
if(str[0]=='A') update(a, b, 1, n, 1) ;
else if(str[0]=='S') update(a, -b, 1, n, 1) ;
else printf("%d\n", query(a, b, 1, n, 1)) ;
}
}
return 0 ;}