HDU - 1166 敌兵布阵 线段树区间查询

题目链接：https://vjudge.net/problem/HDU-1166

题意：给出每个营地有多少人，执行三种操作：

(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个营地的总人数;

分析：如果直接模拟，暴力求解，肯定会超时。由于操作繁多，有修改，有查询，肯定可以用线段树求解。确定类型属于单点更新，区间查询。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#define L(u) (u<<1)
#define R(u) (u<<1|1)
using namespace std;
const int maxn = 50010;
int a[maxn];

struct Node {
    int l,r;
    int sum;
} node[maxn<<2];

void Pushup(int u) { //向上延时更新
    node[u].sum=node[L(u)].sum+node[R(u)].sum;
    return;
}

void Build(int u,int left,int right) { //建树
    node[u].l=left;
    node[u].r=right;
    if(node[u].l==node[u].r) {
        node[u].sum=a[left];
        return;
    }
    int mid=(node[u].l+node[u].r)>>1;
    Build(L(u),left,mid);
    Build(R(u),mid+1,right);
    Pushup(u);
}

void Update(int u,int pos,int val) { //更新结点
    if(node[u].l==node[u].r) {
        node[u].sum+=val;
        return;
    }
    int mid=(node[u].l+node[u].r)>>1;
    if(pos<=mid) Update(L(u),pos,val);
    else Update(R(u),pos,val);
    Pushup(u);
}

int Quary(int u,int left,int right) { //区间查询
    if(left==node[u].l&&right==node[u].r) return node[u].sum;
    int mid=(node[u].l+node[u].r)>>1;
    if(right<=mid) return Quary(L(u),left,right);
    else if(left>mid) return Quary(R(u),left,right);
    else return Quary(L(u),left,mid)+Quary(R(u),mid+1,right);
}

int main () {
    //freopen("in.txt","r",stdin);
    int T,Cas=0;
    scanf("%d",&T);
    while(T--) {
        printf("Case %d:\n",++Cas);
        int n,p,add,x,y;
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        Build(1,1,n);
        char s[20];
        while(1) {
            scanf("%s",s);
            if(strcmp(s,"Add")==0) {
                scanf("%d%d",&p,&add);
                Update(1,p,add);
            } else if(strcmp(s,"Sub")==0) {
                scanf("%d%d",&p,&add);
                Update(1,p,-add);
            } else if(strcmp(s,"Query")==0) {
                scanf("%d%d",&x,&y);
                printf("%d\n",Quary(1,x,y));
            } else break;
        }
    }
    return 0;
}