题意:
有N个兵营,每个兵营都给出了人数ai(下标从1开始),有四种命令
(1)”Addij",表示第i个营地增加j人。
(2)“Sub i j”,表示第i个营地减少j人。
(3)“Query ij",查询第i个营地到第j个营地的总人数。
(4)”End“,表示命令结束。
分析:
一道极水的线段树,单点修改乱搞就好了。
代码:
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1001000;
int n,m,ans,x,z,l,r,cas,a[N];
struct tree{int l,r,sum;}g[N];
char s[1000];
void build(int rt,int l,int r)
{
g[rt].l=l;
g[rt].r=r;
g[rt].sum=0;
if (l==r) g[rt].sum=a[l]; else
{
int mid=(l+r)/2;
build(2*rt,l,mid);
build(2*rt+1,mid+1,r);
g[rt].sum=g[2*rt].sum+g[2*rt+1].sum;
}
}
void update(int rt,int k,int val)
{
int l=g[rt].l,r=g[rt].r;
if (l==r) g[rt].sum+=val; else
{
int mid=(l+r)/2;
if (k<=mid)
update(2*rt,k,val); else
update(2*rt+1,k,val);
g[rt].sum=g[2*rt].sum+g[2*rt+1].sum;
}
}
int query(int rt,int ll,int rr)
{
int l=g[rt].l,r=g[rt].r;
if (ll<=l && r<=rr) return g[rt].sum;
int mid=(l+r)/2,suml=0,sumr=0;
if (ll<=mid) suml=query(2*rt,ll,rr);
if (rr> mid) sumr=query(2*rt+1,ll,rr);
return suml+sumr;
}
int main()
{
int T;
scanf("%d",&T);
while (T--)
{
printf("Case %d:\n",++cas);
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",a+i);
build(1,1,n);
while (scanf("%s",s) && s[0]!='E')
{
if (s[0]=='Q')
{
scanf("%d%d",&l,&r);
printf("%d\n",query(1,l,r));
} else
if (s[0]=='S')
{
scanf("%d%d",&x,&z);
update(1,x,-z);
} else
{
scanf("%d%d",&x,&z);
update(1,x,z);
}
}
}
return 0;
}