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

题目链接: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;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值