【HDU1166】敌兵布阵

线段树模板题

#include<cstdio>
#include<cstring>
using namespace std;
const int N=50010;
int t,n,h[N],sum[4*N];
char ch[20];
void update(int),build(int,int,int),
  modify(int,int,int,int,int),work();
int query(int,int,int,int,int),turn();
int main(){
    scanf("%d",&t);
    for (int k=1;k<=t;k++){
        memset(sum,0,sizeof(sum));
        printf("Case %d:\n",k);
        work();
    }
    return 0;
}
int turn(){
    if (ch[0]=='Q') return 1;
    if (ch[0]=='E') return 0;
    if (ch[0]=='A') return 2;
    return 3;
}
void work(){
    int a,b,c;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
        scanf("%d",&h[i]);
    build(1,1,n);
    getchar();scanf("%s",ch);
    c=turn();
    while (c){
        scanf("%d %d",&a,&b);
        if (c==1)
            printf("%d\n",query(1,1,n,a,b));
        else{
            if (c==3) b*=-1;
            modify(1,1,n,a,b);
        }
        getchar();scanf("%s",ch);
        c=turn();
    }
}
void update(int i){
    sum[i]=sum[i<<1]+sum[i<<1|1];
}
void build(int i,int l,int r){
    if (l==r){
        sum[i]=h[l];return;
    }
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
    update(i);
}
int query(int i,int l,int r,int L,int R){
    if (L<=l&&r<=R) return sum[i];
    int mid=(l+r)>>1,ans=0;
    if (L<=mid) ans+=query(i<<1,l,mid,L,R);
    if (R>mid) ans+=query(i<<1|1,mid+1,r,L,R);
    return ans;
}
void modify(int i,int l,int r,int x,int d){
    if (l==r){
        sum[i]+=d;return;
    }
    int mid=(l+r)>>1;
    if (x<=mid) modify(i<<1,l,mid,x,d);
    else modify(i<<1|1,mid+1,r,x,d);
    update(i);
}