HDU 1166模仿大牛写的线段树

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

HDU 1166

十分优美;

上代码 以后多记着

#include<stdio.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define maxn 55555
char buffer[10];
int sum[maxn<<2];
int PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
int build(int l,int r,int rt)
{
    if(l==r) {scanf("%d",&sum[rt]);return 0;}
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}
int updata(int p,int k,int l,int r,int rt)
{
    int m;
    if(l==r) {sum[rt]+=k;return 0;}
    m=(l+r)>>1;
    if(p<=m) updata(p,k,lson);
    else updata(p,k,rson);
    PushUp(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    int temp=0,m;
    if(L<=l&&r<=R) return sum[rt];
    m=(l+r)>>1;
    if(L<=m) temp=temp+query(L,R,lson);
    if(R>m) temp=temp+query(L,R,rson);
    return temp;
}
int main()
{
    freopen("hdu1166.in","r",stdin);
    freopen("hdu1166.out","w",stdout);
    int T,i,N,a,b;
    scanf("%d",&T);
    for(i=1;i<=T;i++)
        {
            scanf("%d",&N);
            build(1,N,1);
            printf("Case %d:\n",i);
            while(scanf("%s",buffer))
                {
                    if(buffer[0]=='E') break;
                    scanf("%d%d",&a,&b);
                    if(buffer[0]=='A') updata(a,b,1,N,1);
                    else if(buffer[0]=='S') updata(a,-b,1,N,1);
                    else if(buffer[0]=='Q') printf("%d\n",query(a,b,1,N,1));
                }
        }
}