题意:n个敌兵的阵营,多个操作:add a ,x:第a个阵营增加x个人;sub a,x:第a个阵营减少x;query a,b:查询区间[a,b]的总人数。
分析:
单点更新,区间查询。直接写。
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=50000;
int t,n;
ll sum[maxn<<2+10];
void pushup(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void creat(int l,int r,int rt)
{
if(l==r){
scanf("%d",&sum[rt]);
return;
}
int mid=(l+r)>>1;
creat(l,mid,rt<<1);
creat(mid+1,r,rt<<1|1);
pushup(rt);
}
void upd(int a,int x,int l,int r,int rt)
{
if(l==r){
sum[rt]+=x;
return;
}
int mid=(l+r)>>1;
if(a<=mid) upd(a,x,l,mid,rt<<1);
else upd(a,x,mid+1,r,rt<<1|1);
pushup(rt);
}
ll query(int a,int b,int l,int r,int rt)
{
if(a<=l&&b>=r) return sum[rt];
int mid=(l+r)>>1;
ll ret;
if(a<=mid) ret=query(a,b,l,mid,rt<<1);
if(b>mid) ret+=query(a,b,mid+1,r,rt<<1|1);
return ret;
}
int main()
{
scanf("%d",&t);
int cas=1;
while(t--){
scanf("%d",&n);
creat(1,n,1);
printf("Case %d:\n",cas++);
char q[20];
while(scanf("%s",q)!=EOF){
if(q[0]=='E') break;
ll a,b;
scanf("%d%d",&a,&b);
if(q[0]=='A'||q[0]=='S'){
if(q[0]=='S') b*=-1;
upd(a,b,1,n,1);
}
else printf("%I64d\n",query(a,b,1,n,1));
}
}
}