hdu 1166 敌兵布阵
2015-07-16 22:11
225 查看
题意 : 敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
接下来每行有一条命令,命令有4种形式:
(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 表示结束,这条命令在每组数据最后出现;
这题显而易见,是典型的线段树,Add和Sub命令本质上是同一个命令,即线段树的单点更新,Q命令就是求i到j这个区间的和。没什么坑点。
代码:
接下来每行有一条命令,命令有4种形式:
(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 表示结束,这条命令在每组数据最后出现;
这题显而易见,是典型的线段树,Add和Sub命令本质上是同一个命令,即线段树的单点更新,Q命令就是求i到j这个区间的和。没什么坑点。
代码:
#include <cstdio> #include <cstring> int sum[50000 << 2]; int n; void build(int rt,int l, int r) { if(l == r) { scanf("%d", &sum[rt]); //printf("%d %d %d\n",l,rt,sum[rt]); return; } build(rt <<1 , l, (l+r)>> 1); build((rt << 1) + 1, ((l+r) >> 1) + 1 , r); sum[rt] = sum[rt << 1] + sum[(rt << 1) + 1]; //printf("%d %d\n", rt, sum[rt]); } void update(int p, int add, int l, int r, int rt) { //printf("%d %d\n", l, r); if(l == r) { //printf("%d\n",l); sum[rt] += add; return; } int mid = (l + r) >> 1; //printf("%d\n",mid); if (p <= mid) update(p, add , l, mid, rt << 1); else update(p, add , mid + 1, r, (rt << 1) + 1); sum[rt] = sum[rt << 1] + sum[(rt << 1) +1]; } int query(int L, int R, int l, int r, int rt) { //printf("%d %d\n",l,r); if(L <= l && R >= r) //当前区间在所求区间内 { //printf("%d\n",sum[rt]); return (sum[rt]); } int mid = (l + r) >> 1; int ret = 0; if(L <= mid) ret += query(L, R, l, mid, rt << 1); //printf("%d %d %d\n",l,r,ret); if(R > mid) ret += query(L, R, mid+1, r, (rt << 1)+1); //printf("%d %d %d\n",l,r,ret); return ret; } int main() { int T; scanf("%d",&T); for(int i=1;i<=T;i++) { printf("Case %d:\n",i); scanf("%d", &n); build(1,1,n); char s[10]; while(scanf("%s",s),s[0]!='E') { int a,b; scanf("%d %d",&a ,&b); if(s[0] == 'Q') { printf("%d\n",query(a, b, 1, n, 1)); } else if(s[0] == 'A') { update(a, b, 1, n, 1); } else if(s[0] == 'S') update(a, -b, 1, n, 1); } } return 0; }