题意:给出一列数,之后做四种操作中的一种
(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个营地的总人数;
(4)End 表示结束,这条命令在每组数据最后出现;
每组数据最多有40000条命令
思路:先建立线段树,之后相应的进行点修改,query的时候递归分段求就行了
代码:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#define N 50005
using namespace std;
int n;
int p,q;
struct Node
{
int sum,l,r;
};
Node node[N<<2];
int num[N];
void built(int t,int l,int r)
{
node[t].l=l;
node[t].r=r;
if(l==r) {
node[t].sum=num[l];
return;
}
int mid=(l+r)>>1;
built(t<<1,l,mid);
built(t<<1|1,mid+1,r);
node[t].sum=node[t<<1].sum+node[t<<1|1].sum;
}
void add(int t,int l,int r)
{
int mid=(l+r)>>1;
if(l==r) node[t].sum+=q;
else{
if(p<=mid) add(t<<1,l,mid);
else add(t<<1|1,mid+1,r);
node[t].sum=node[t<<1].sum+node[t<<1|1].sum;
}
}
int query(int t,int l,int r)
{
int mid=(node[t].l+node[t].r)>>1,ans=0;
if(node[t].l==l && node[t].r==r) return node[t].sum;
if(r<=mid) ans+=query(t<<1,l,r);
else if(l>mid) ans+=query(t<<1|1,l,r);
else{
ans+=query(t<<1,l,mid);
ans+=query(t<<1|1,mid+1,r);
}
return ans;
}
int main()
{
int t,aaa=1;
char s[10];
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(node,0,sizeof(node));
for(int i=1;i<=n;i++) scanf("%d",&num[i]);
built(1,1,n);
printf("Case %d:\n",aaa++);
while(scanf("%s",s) && strcmp(s,"End"))
{
scanf("%d %d",&p,&q);
if(s[0]=='A')
{
add(1,1,n);
}
else if(s[0]=='S')
{
q=-q;
add(1,1,n);
}
else
{
int ans=query(1,p,q);
printf("%d\n",ans);
}
}
}
return 0;
}