您的位置:首页 > 其它

敌兵布阵-线段树(1)

2015-08-25 16:55 134 查看
A - 敌兵布阵
Time Limit:1000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u
Submit Status

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

是线段树的模板题
把区间设为自己到自己!

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 100010<<2//一棵树的某排有N个数,那此排以前就一共有N-1个数,且此排的下一排有2N个数,N+N-1+2N约等于4N,要开N的四倍数组,才不会越界
int sto
,up
,x,y;
char s[3]; 8 void init(int t,int w,int o)//初始化树中的某一结点,t为该结点左边界,w为右边界,o为该结点的标号(用来表示该结点的位置)
{
up[o]=0;//up[o]代表该结点的值需要改变时,增加或减少的量;
if(t==w)//当递归到最底层,即划分到区间长度为0时
{
scanf("%d",&sto[o]);//输出该结点的数值
return;
}
int mid=(t+w)>>1;//除以2,划分下一层的边界
int lso=o<<1;//乘以二,求出其左子数的标号
init(t,mid,lso);//递归到其左子树
init(mid+1,w,lso|1);
sto[o]=sto[lso]+sto[lso|1];//该结点的左子树的值加上右子树的值,从下到上进行累加,获得整棵数的值得累加和
}
//某个结点代表的是某个范围,此结点想要变化N,就代表这一范围的每个数都要变化N;
void updown(int sum,int o)//更新某个结点的值的操作,sum指该结点以下有多少个结点(他们同样要一起更新),o代表该结点的标号
{//第一次修改值得时候,是不会用到的,因为会有多组测试数据,会进行多次修改,所以才要更新
if(up[o])//如果变化量不为0,则需要进行更新
{
int lso=o<<1;//此结点的左子树标号
up[lso]+=up[o];//把变化量传给左子数
up[lso|1]+=up[o];//把变化量传给右子数
sto[lso]+=(sum-(sum>>1))*up[o];//左子树的值为它自身加上它以下的所有结点数变化量之和(左子树的值要多1,所以要加多的那一边)
sto[lso|1]+=(sum>>1)*up[o];
up[o]=0;//用完了就制为0,以免后面重复加
}
}
int updata(int t,int w,int o,int st,int en,int num)//查找并计算出某颗树的值,t,w为整棵树的范围,o,代表查找的位置,st,en,代表该位置的范围,num代表变化量
{
if(st<=t&&w<=en)//当查到那个那个位置的范围以内时
{
if(num!=1<<29)//如果num不为无穷大,即除了查询操作,还要进行更新工作
{
up[o]+=num;//把要加的变量付给up[o]
sto[o]+=(w-t+1)*num;//算出该范围总的变化量
}
return sto[o];//返回求和
}
updown(w-t+1,o);//继续向下更新
int mid=(t+w)>>1;//继续缩小范围
int lso=o<<1;
int ans=0;
if(st<=mid) ans+=updata(t,mid,lso,st,en,num);//当st小于mid 则表明该范围至少有一部分在左子树;则减小范围,再次进行查找计算,然后一层一层退回;
if(en>mid) ans+=updata(mid+1,w,lso|1,st,en,num);//当en>mid 说明,该范围至少有一部分在右子树,则减小范围,再次进行查找计算,然后层层退回;
sto[o]=sto[lso]+sto[lso|1];//求出每一个范围的总值;
return ans;//每次返回回去都不再重新走ans=0,ans一直累加,直到输出答案
}
int main()
{
int n,m,t=1;
scanf("%d",&n);
while(n--)
{
scanf("%d",&m);
init(1,m,1);
printf("Case %d:\n",t++);
while(1)
{
scanf("%s",s);
if(s[0]=='Q')
{
scanf("%d%d",&x,&y);
printf("%d\n",updata(1,m,1,x,y,1<<29));//把查询范围x到y的和为多少
}
if(s[0]=='A')
{
scanf("%d%d",&x,&y);
updata(1,m,1,x,x,y);//把范围订在自己到自己
}
if(s[0]=='S')
{
scanf("%d%d",&x,&y);
updata(1,m,1,x,x,-y);
}
if(s[0]=='E')
break;
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: