HDU 1166 敌兵布阵 线段树求解

题目链接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66989#problem/A

题意：动态修改某个点的值，然后动态询问区间和。

标准的线段树点修改问题，以前做过，第一次看着别人的代码用线段树水过了，后来还用树状数组做过（现在不记得树状数组了。。。。），现在再次用线段树写，当然这次是独立写的。题意本身不难，但是有一个值的注意，本题输入与输出量比较大，得用scanf与printf，否则果断TLE。

代码：

#include<iostream>
#include<cstdio>
#define maxn 50010

using namespace std;
struct node{
    int l,r,w;
}T[maxn*3];

int N,A[maxn];//A数组要不要都行，初始时，每输入一个值调用一次add
char s[10];

int build(int L,int R,int o){
    T[o].l=L;T[o].r=R;
    if(L==R) return T[o].w=A[L];
    return T[o].w=build(L,(L+R)/2,o*2)+build((L+R)/2+1,R,o*2+1);
}

void add(int L,int R,int val,int o){
    if(T[o].l>=L && T[o].r<=R){
        T[o].w+=val;return;
    }
    if(T[o].l==T[o].r) return;
    T[o].w+=val;
    int tl=o*2,tr;tr=tl+1;
    if(L>=T[tr].l) add(L,R,val,tr);
    else if(R<=T[tl].r) add(L,R,val,tl);
    else{
        add(L,T[tl].r,val,tl);
        add(T[tr].l,R,val,tr);
    }
}

int query(int L,int R,int o){
    if(T[o].l>=L && T[o].r<=R) return T[o].w;
    if(T[o].l==T[o].r) return 0;
    int tl=o*2,tr;tr=tl+1;
    if(L>=T[tr].l) return query(L,R,tr);
    if(R<=T[tl].r) return query(L,R,tl);
    return query(L,T[tl].r,tl)+query(T[tr].l,R,tr);
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    int T;cin>>T;
    for(int kase=1;kase<=T;kase++){
        cin>>N;
        for(int i=0;i<N;i++)
            scanf("%d",A+i+1);
        build(0,N,1);
        printf("Case %d:\n",kase);
        while(scanf("%s",s) && s[0]!='E'){
            int a,b;
            scanf("%d %d",&a,&b);
            if(s[0]=='Q') printf("%d\n",query(a,b,1));
            else if(s[0]=='A') add(a,a,b,1);
            else add(a,a,0-b,1);//减法发也可以是加法
        }
    }
    return 0;
}