线段树区间合并主要解决一段连续区间修改,查询问题。
线段树是树形结构,为解决相邻区间更新,修改问题,我们必须在原本须要维护的区间内,连续段最大值的基础上加上,从左,右两边起,最大的连续区间长度。
例题:
酒店:
此题开始时,所有的房间都是空房间,有两个操作:
1、查询长度为x的连续房间,并将房间全部填满,输出左端点值。
2、表示退房,将x,到x+y-1的房间退房。
我们来看线段树,我们知道线段树要将区间分成好多块,之后对每一块进行处理。对于两块深度相同的区间,如果只是单纯的维护一个区间最大值,很明显。对于两个区间共同的祖宗区间,你无法判断,祖宗区间的最大值,是它左儿子的最大值还是右儿子的最大值,还是这个最大的连续区间在两个区间之间。因此引入两个新的值,这个区间从最左段起最大的连续区间,还有从这个区间最右端起最大的连续区间,这样我们就可以判断祖宗节点最大的连续区间到底是多大。
向上更新:
对于我们维护的三个值,左最大跟右最大的更新几乎是一样的:
(图比较丑,emm,手残多见谅。)
我们看到祖宗节点的左最大为1,就是左儿子的左最大,右最大为4,我们单看右儿子最大只为3,但是左儿子还有一个点与它们相邻,因此当右儿子右最大等于它所包含的区间长度时,那么我们就可以将左儿子右最大也更新到祖宗节点中,具体见代码。
向下更新,对于lazy节点我们只需要让他记录是开房,还是退房,记录完毕向下更新,就将区间全部清空,或者全部住房即可。
对于查询,我们可以看到,需要查找长度为x的区间,输出的是最左端的房间,也就是说有多个空房时,我们只要输出最左端的那一个,那么我们的查找只要从最左端找起,如果有就输出没有就继续查找中间,最后查找最右端。,找到最左端值后,我们再根据区间长度调用更新函数,区间更新即可。
代码:
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#define ll long long
const int M=1e5+1000;
using namespace std;
//对于入住情况,先查询满足情况的左端点。然后根据左端点,求出更新区间,对整个区间进行更新
struct node
{
ll sum;
ll lmax;//从最左边有多少个空房间
ll rmax;//从最右边有多少个空房间
int lazy;//延迟更新,如果入住为1,退房为2
}tr[M*4];
void build(int i,int l,int r)
{
tr[i].sum=tr[i].lmax=tr[i].rmax=r-l+1;
tr[i].lazy=0;
if(l==r)
{
return;
}
int mid=(l+r)/2;
build(i*2,l,mid);
build(i*2+1,mid+1,r);
}
void pushup(int i,int l,int r)
{
int mid=(r+l)/2;
if(tr[i*2].sum==(mid-l+1))
{
tr[i].lmax=tr[i*2].sum+tr[i*2+1].lmax;
}
else
tr[i].lmax=tr[i*2].lmax;
if(tr[i*2+1].sum==(r-mid))
{
tr[i].rmax=tr[i*2+1].sum+tr[i*2].rmax;
}
else
tr[i].rmax=tr[i*2+1].rmax;
tr[i].sum=max(tr[i*2].rmax+tr[i*2+1].lmax,max(tr[i*2].sum,tr[i*2&