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;
}