hdu1166(线段树单点修改)

链接：点击打开链接

题意：第一行一个整数T，表示有T组数据.
每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50).
接下来每行有一条命令,命令有4种形式:
(1)Add(i,j) i和j为正整数,表示第i个营地增加j个(j不超过30)
(2)Sub(i,j) i和j为正整数,表示第i个营地减少j个人(j不超过30)
(3)Query(i,j),i和j为正整数,i<=j,表示询问第i到第j个营地的总人数
(4)End表示结束,这条命令在每组数据最后出现
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内

代码：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int siz=50005;
int tree[siz<<2];
void build(int l,int r,int rt){
    if(l==r){
        scanf("%d",&tree[rt]);
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void Change(int p,int q,int l,int r,int rt){
    if(l==r){
        tree[rt]+=q;
        return;
    }
    int m=(l+r)>>1;
    if(p<=m)
    Change(p,q,l,m,rt<<1);
    else
    Change(p,q,m+1,r,rt<<1|1);
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
int query(int L,int R,int l,int r,int rt){
    if(L<=l&&r<=R)
    return tree[rt];
    int ans=0,m=(l+r)>>1;
    if(L<=m)
    ans+=query(L,R,l,m,rt<<1);
    if(R>m)
    ans+=query(L,R,m+1,r,rt<<1|1);
    return ans;
}
int main(){                                     //线段树单点修改模板题
    string s;
    int t,n,i,j,u,v,cas;
    scanf("%d",&t);
    for(cas=1;cas<=t;cas++){
        scanf("%d",&n);
        build(1,n,1);
        printf("Case %d:\n",cas);
        while(cin>>s&&s!="End"){
            scanf("%d%d",&u,&v);
            if(s=="Query")
            printf("%d\n",query(u,v,1,n,1));
            else if(s=="Add")
            Change(u,v,1,n,1);
            else
            Change(u,-v,1,n,1);
        }
    }
    return 0;
}