HDU1166敌兵布阵 线段树模板题

题目：传送门

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<map>
#include<queue>
#include<cmath>
#include<stack>
#include<vector>
#include<cstdio>
#define MAXN 50000
#define INF 1000000007
using namespace std;
int sum[MAXN<<2];
int A[MAXN];
int max(int a,int b){if(a>=b)return a;else return b;}
void pushup(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
if(l==r)
{
sum[rt]=A[l];
return ;
}
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
pushup(rt);
}
void update(int L,int C,int l,int r,int rt)
{
if(l==r)
{
sum[rt]+=C;
return ;
}
int m=(l+r)>>1;
if(L<=m) update(L,C,l,m,rt<<1);
else update(L,C,m+1,r,rt<<1|1);
pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
if(L<=l&&R>=r)
{
return sum[rt];
}
int m=(l+r)>>1;
int ans=0;
if(L<=m) ans+=query(L,R,l,m,rt<<1);
if(R>m) ans+=query(L,R,m+1,r,rt<<1|1);
return ans;
}
int main()
{
int firts=1;
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&A[i]);
build(1,n,1);
char s[10];
int a,b;
cout<<"Case "<<firts++<<":"<<endl;
while(scanf("%s",s)==1)
{
if(strcmp(s,"End")==0)
break;
scanf("%d%d",&a,&b);
if(s[0]=='Q')
{
cout<<query(a,b,1,n,1)<<endl
4000
;
continue;
}
if(s[0]=='A')
{
update(a,b,1,n,1);
continue;
}
if(s[0]=='S')
{
update(a,-b,1,n,1);
continue;
}
}
}
return 0;
}