您的位置:首页 > 其它

HDU 1166 敌兵布阵 线段树

2018-01-17 22:00 357 查看
[align=left]Problem Description[/align]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的责骂的.
 
[align=left]Input[/align]第一行一个整数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条命令
 
[align=left]Output[/align]对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。
 
[align=left]Sample Input[/align]
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  
[align=left]Sample Output[/align]
Case 1:
6
33
59

作为一个老队员竟然第一次写线段树,哎,老队员集训的专题一个接一个的,已经讲了BSGS、数位DP、FFT,这些随便一个我都够呛,BSGS只理解了大概,FFT是今天的内容完全不懂,暂时放弃,直接准备下面的专题。接下来三个都是线段树的专题,然而我连线段树的题都从来没刷过,于是放弃FFT空出时间来学学线段树。
虽然是一道很简单的线段树,但是写出来这道题还是有点成就感的,毕竟是第一道线段树。也正是这点成就感才能让我们搞acm的坚持下去。
不懂线段树的可以去这里先学习学习。
***|的优先级比<<的优先级低,+优先级比<<优先级高,所以
[cpp] view plain copynode<<1+1等价于node<<(1+1)  
node<<1|1等价于(node<<1)+1

#include<bits/stdc++.h>
using namespace std;
int a[50000+10];
char str[10];
struct node
{
int l,r,sum;//当前节点下面的数的序号范围l-r,这些数的和
}tree[4*50000+10];

void build(int l,int r,int node)
{
tree[node].l=l,tree[node].r=r;
if(l==r)
{
tree[node].sum=a[l];
return ;
}
int mid=(l+r)>>1;
build(l,mid,node<<1);
build(mid+1,r,node<<1|1);
tree[node].sum=tree[node<<1].sum+tree[node<<1|1].sum;
}
void updat(int a,int val,int node)
{

if(tree[node].l==a&&tree[node].r==a//每一个数都保存在二叉树的一片叶子上,当节点没有儿子的时候就是叶子
{
tree[node].sum+=val;
return ;
}
int mid=(tree[node].l+tree[node].r)>>1;
if(a<=mid)//判断在左边还是右边
{
updat(a,val,node<<1);
}
else
updat(a,val,node<<1|1);
tree[node].sum=tree[node<<1].sum+tree[node<<1|1].sum;//回溯更新祖先的节点上sum的值
}
int cal(int l,int r,int node)
{
// printf("%d %d %d\n",node,tree[node].l,tree[node].r);
if(l==tree[node].l&&r==tree[node].r)//当前区间是需要寻找的区间的子集,直接加上该区间的值
return tree[node].sum;
int mid=(tree[node].l+tree[node].r)>>1;
if(r<=mid)//当前寻找的区间在当前区间左边
{
return cal(l,r,node<<1);
}
else if(l>mid)//当前寻找的区间在当前区间右边
return cal(l,r,node<<1|1);
return cal(l,mid,node<<1)+cal(mid+1,r,node<<1|1);//当前寻找的区间在当前区间中间,则将寻找的区间拆分
}
int main()
{
int t;
scanf("%d",&t);
// memset(tree,-1,sizeof(int));
for(int kase=1;kase<=t;kase++)
{
printf("Case %d:\n",kase);
int n;
// memset(s,0,sizeof(s));
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
build(1,n,1);
// for(int i=1;i<22;i++)
// printf("%d %d %d %d\n",i,tree[i].l,tree[i].r,tree[i].sum);
while(scanf("%s",str))
{
if(str[0]=='E')
break;
int x,y;
scanf("%d%d",&x,&y);
if(str[0]=='A')
{
updat(x,y,1);
}
else if(str[0]=='S')
updat(x,-y,1);
else
printf("%d\n",cal(x,y,1));
}

}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: