题目链接:点击打开链接
大意:敌人有N个营地,一开始知道每个营地的人数。
接下来会询问四种命令:(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 表示结束,这条命令在每组数据最后出现。
遇到Query则作出回答,遇到End则结束本次数据的询问。其他则修改相关营地的数据。(显然,Sub 可以归并到 Add 操作)
思路:
对于每一组数据,用build初始化线段树,每个结点保存区间[i, j]内的总人数。叶子结点即是每个营地的人数,这个由输入给出,对于内结点,则采取pushUp由下往上传递值。
对于Add 操作(包括Sub操作),可以用update来修改叶子结点的值,与build相似,内结点采取pushUp更新其值。
对于Query 操作,用query来查看区间的人数。对于没有在一个存储区间表示出来的空间,则采取同时向两边走,覆盖后返回其值的方式取值,详细见代码。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int maxn = 50000 + 10;
int sum[maxn<<2];
void pushUp(int rt)
{
sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l, int r, int rt)
{
if(l == r)
{
scanf("%d", &sum[rt]);
return;
}
int m = (l+r)>>1;
build(lson);
build(rson);
pushUp(rt);
}
void update(int p, int add, int l, int r, int rt)
{
if(l == r)
{
sum[rt] += add;
return;
}
int m = (l+r)>>1;
if(p <= m)
update(p, add, lson);
else
update(p, add, rson);
pushUp(rt);
}
int query(int L, int R, int l, int r, int rt)
{
if(L<=l && r<=R)
return sum[rt];
int m = (l+r)>>1;
int ans = 0;
if(L <= m)
ans += query(L, R, lson);
if(R > m)
ans += query(L, R, rson);
return ans;
}
int main()
{
int T, n, a, b;
char sa[10];
scanf("%d", &T);
for(int i = 1; i <= T; i++)
{
printf("Case %d:\n", i);
scanf("%d", &n);
build(1, n, 1);
while(scanf("%s", sa), sa[0] != 'E')
{
scanf("%d%d", &a, &b);
if(sa[0] == 'Q')
printf("%d\n",query(a, b, 1, n, 1));
else if(sa[0] == 'A')
update(a, b, 1, n, 1);
else
update(a, -b, 1, n, 1);
}
}
return 0;
}