hdu 1166

来源:互联网 发布:jquery储存数据 编辑:程序博客网 时间:2024/06/07 15:48

题意:

有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); elseupdate(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));} elseif (s[0]=='S'){scanf("%d%d",&x,&z);update(1,x,-z);} else{scanf("%d%d",&x,&z);update(1,x,z);}}}return 0;}


0 0
原创粉丝点击