HDU_1166 敌兵布阵

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

最最基础的线段树,只更新叶子节点,然后把信息用PushUP(int r)这个函数更新上来

线段树功能:update:单点增减 query:区间求和

#include<cstdio>
#define lson l , m ,rt << 1
#define rson m+1 , r , rt << 1 | 1
const int maxn = 55555;
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,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') update(a , -b , 1 , n , 1);
            else update(a , b , 1 , n , 1);
        }
    }
    return 0;
}