hdu 1166 敌兵布阵 线段树

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>

#define lson l, m, rt << 1
#define rson m+1, r, rt << 1 | 1

using namespace std;

const int MAXN = 50010;
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 UFset(int p, int a, int l, int r, int rt)
{
if(l == r)
{
sum[rt] += a;
return;
}
int m = (l+r) >> 1;
if(p <= m)
UFset(p, a, lson);
else
UFset(p, a, 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 , n;
scanf("%d",&T);
for (int cas = 1 ; cas <= T ; cas ++)
{
printf("Case %d:\n",cas);
scanf("%d",&n);
build(1 , n , 1);
char op[10];
while (scanf("%s",op))
{
if (op[0] == 'E')
break;
int a , b;
scanf("%d%d",&a,&b);
if (op[0] == 'Q')
printf("%d\n",Query(a , b , 1 , n , 1));
else if (op[0] == 'S')
UFset(a , -b , 1 , n , 1);
else
UFset(a , b , 1 , n , 1);
}
}
return 0;
}