题目链接: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];
else
return 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|1
const 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];
else
return 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;
}