hdu 1540 Tunnel Warfare(线段树)(第三部分 区间合并)

来源:互联网 发布:python网络编程 pdf 编辑:程序博客网 时间:2024/05/19 04:05

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1540

题目大意:

D a代表破坏村庄a;

R代表修复最后被破坏的那个村庄;

Q代表询问包括x在内的最大连续区间是多少;

思路:

一:存线段树数据的数组要有四个变量。

lsum 存的是从本节点区间最左端开始(向右)一共有多少个连续的空房间

rlsum 存的是 从本节点区间最右端开始(向左)一共有多少个连续的空房间,

msum存的是 本区间一共最多有 多少个连续的空房间,

pre 存的是摧毁的村庄(按顺序,便于之后的修护)

二:关于pushup函数

void pushup(int rt,int m)//向上传递{lsum[rt]=lsum[rt<<1];rsum[rt]=rsum[rt<<1|1];if(lsum[rt]==m-(m>>1))lsum[rt]+=lsum[rt<<1|1];if(rsum[rt]==(m>>1))rsum[rt]+=rsum[rt<<1];msum[rt]=max(msum[rt<<1],msum[rt<<1|1]);msum[rt]=max(msum[rt],rsum[rt<<1]+lsum[rt<<1|1]);}
就是当更新树中一节点后  回溯更新该节点的父节点的  lsum   rsum  msum  sum的值。。

我们可以发现 节点 rt 的lsum rsum msum 值是与它左右儿子节点的lsum  rsum  msum值有关系的

 rt 节点的lsum 值  等于 他左儿子的lsum值 ,如果左儿子区间整个序列都是连续的,那么lsum的值等于 左儿子的lsum值 + 右儿子的lsum值  具体参考代码理解

同理:rsum值也一样……

对于 rt 节点 的msum值等于Max( 左儿子的msum值,右儿子的msum值 ,左儿子的rsum值+右儿子的lsum值)

三:关于query函数

int query(int p,int l,int r,int rt){if(l==r||msum[rt]==0||msum[rt]==r-l+1)//到了叶子节点或者该访问区间为空或者已满都不必要往下走了{return msum[rt];}int m=(r+l)>>1;if(p<=m)//p在其左子树{if(m-p+1<=rsum[rt<<1])//判断p是否在左子树的右区间内  如果在,则需要加上右子树的左区间{return query(p,lson)+lsum[rt<<1|1];}else{return query(p,lson);}}if(p>m)//p在其右子树{if(p-m<=lsum[rt<<1|1])//判断p是否在右子树的左区间内  如果在,则需要加上左子树的右区间return query(p,rson)+rsum[rt<<1];elsereturn query(p,rson);}}
这里需要理解是一系列的判断:首先判断他在左子树还是右子树,加入他在左子树,那么需要判断p是否在左子树的右区间内 ,如果在,则需要加上右子树的左区间

对于在右子树的话同理

接下来直接上AC代码:

#include<iostream>#include<cmath>#include<cstdio>#include<iomanip>#include<cstring>#include<stdlib.h>#include<string>#include<algorithm>#include<queue>using namespace std;#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1const int  maxn =60050;int top;int lsum[maxn<<2],rsum[maxn<<2],msum[maxn<<2];int pre[maxn<<2];void pushup(int rt,int m)//向上传递{lsum[rt]=lsum[rt<<1];rsum[rt]=rsum[rt<<1|1];if(lsum[rt]==m-(m>>1))lsum[rt]+=lsum[rt<<1|1];if(rsum[rt]==(m>>1))rsum[rt]+=rsum[rt<<1];msum[rt]=max(msum[rt<<1],msum[rt<<1|1]);msum[rt]=max(msum[rt],rsum[rt<<1]+lsum[rt<<1|1]);}void build(int l,int r,int rt){if(l==r){lsum[rt]=rsum[rt]=msum[rt]=1;return ;}int m=(l+r)>>1;build(lson);build(rson);pushup(rt,r-l+1);}int query(int p,int l,int r,int rt){if(l==r||msum[rt]==0||msum[rt]==r-l+1)//到了叶子节点或者该访问区间为空或者已满都不必要往下走了{return msum[rt];}int m=(r+l)>>1;if(p<=m)//p在其左子树{if(m-p+1<=rsum[rt<<1])//判断p是否在左子树的右区间内  如果在,则需要加上右子树的左区间{return query(p,lson)+lsum[rt<<1|1];}else{return query(p,lson);}}if(p>m)//p在其右子树{if(p-m<=lsum[rt<<1|1])//判断p是否在右子树的左区间内  如果在,则需要加上左子树的右区间return query(p,rson)+rsum[rt<<1];elsereturn query(p,rson);}}void update(int p,int c,int l,int r,int rt){if(p==l&&p==r){if(c==1)           lsum[rt]= rsum[rt] = msum[rt] = 1;//修复        else            lsum[rt]= rsum[rt] = msum[rt] = 0;//破坏        return ;}int m=(r+l)>>1;if(p<=m)update(p,c,lson);else if(p>m)update(p,c,rson);pushup(rt,r-l+1);}int main(){int n,m;while(~scanf("%d%d",&n,&m)){build(1,n,1);top=0;while(m--){char op[5];int a,b;scanf("%s",op);if(op[0]=='Q')//查询住房{scanf("%d",&a);int p=query(a,1,n,1);printf("%d\n",p);}else if(op[0]=='D'){scanf("%d",&a);pre[top++]=a;update(a,0,1,n,1);}else{if(top>0){b=pre[top-1];top--;update(b,1,1,n,1);}}}}return 0;}



阅读全文
0 0