敌兵布阵
| ||||||
Description | ||||||
C国的死对头A国这段时间正在进行军事演习,所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段,所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动,可能增加或减少若干人手,但这些都逃不过C国的监视。 中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动,而Derek每次询问的段都不一样,所以Tidy不得不每次都一个一个营地的去数,很快就精疲力尽了,Derek对Tidy的计算速度越来越不满:"你个死肥仔,算得这么慢,我炒你鱿鱼!”Tidy想:“你自己来算算看,这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下,Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说:“死肥仔,叫你平时做多点acm题和看多点算法书,现在尝到苦果了吧!”Tidy说:"我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼,这么算他真的会崩溃的,聪明的读者,你能写个程序帮他完成这项工作吗?不过如果你的程序效率不够高的话,Tidy还是会受到Derek的责骂的. | ||||||
Input | ||||||
第一行一个整数T,表示有T组数据。 每组数据第一行一个正整数N(N<=50000),表示敌人有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 表示结束,这条命令在每组数据最后出现; 每组数据最多有40000条命令 | ||||||
Output | ||||||
对第i组数据,首先输出“Case i:”和回车, 对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。 | ||||||
Sample Input | ||||||
1
10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End | ||||||
Sample Output | ||||||
Case 1: 6 33 59 | ||||||
Source | ||||||
Source HDU |
树状数组和线段树很像,但能用树状数组解决的问题,基本上都能用线段树解决,而线段树能解决的树状数组不一定能解决。相比较而言,树状数组效率要高很多。
这里附上一张树装数组的图,便于理解。(此图摘自百度百科)
树装数组注意点:树状数组每一个2的倍数的节点,都存储着这之前所有数的和。而奇数节点则存储其本身。
同时也要知道x&(-x)的含义:,当x为0时结果为0;x为奇数时,结果为1;x为偶数时,结果为x中2的最大次方的因子。 比如x=32,其中2的最大次方因子为2^5,故x&(-x)结果为32;当x=28,其中2的最大次方因子为4,故x & (-x)结果为4。当x=24,其中2的最大次方因子为8,故 x&(-x)结果为8。
上代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int ss=50000;
int Map[ss+50];
int n,x;
int flag=1;
int lowbit(int x)///用于寻找节点
{
return x&(-x);
}
///x&(-x),当x为0时结果为0;x为奇数时,结果为1;x为偶数时,
///结果为x中2的最大次方的因子。 比如x=32,其中2的最大次方因子为2^5,
///故x&(-x)结果为32;当x=28,其中2的最大次方因子为4,
///故x & (-x)结果为4。当x=24,其中2的最大次方因子为8,故 x&(-x)结果为8.
int update(int p,int x)///建立树装数组,也可以更新树装数组节点的值
{
while(p<=n)///坐标不越界
{
Map[p]+=x;///更新节点上的值
p+=lowbit(p);///寻找下一个节点
}
}
///树状数组每一个2的倍数的节点,都存储着这之前所有数的和
///而奇数节点则存储其本身
int js(int p)///计算和用
{
int sum=0;
while(p>0)
{
sum+=Map[p];
p-=lowbit(p);///奇数节点的和等于奇数节点的数加上前一个偶数节点上的数
}
return sum;
}
int main()
{
int t;
cin>>t;
while(t--)
{
memset(Map,0,sizeof(Map));///注意清空数组
cin>>n;
for(int i=1; i<=n; i++)
{
scanf("%d",&x);
update(i,x);///输入的同时建立树装数组
}
// for(int i=1; i<=n; i++)
// {
// printf("%d ",Map[i]);///输出树装数组,检测用。
// }
printf("Case %d:\n",flag);
flag++;
char re[10];
while(~scanf("%s",re))
{
int i,j;
if(re[0]=='E')
{
break;
}
else if(re[0]=='A')
{
scanf("%d %d",&i,&j);
update(i,j);///更新树的节点数据
}
else if(re[0]=='S')
{
scanf("%d %d",&i,&j);
update(i,-j);///更新树的节点数据
}
else if(re[0]=='Q')
{
scanf("%d %d",&i,&j);
printf("%d\n",js(j)-js(i-1));///注意这里的i要减去1,否则会把i节点本身也删去
}
}
}
return 0;
}