题目大意:
1 a:询问是不是有连续长度为a的空房间,有的话住进最左边
2 a b:将[a,a+b-1]的房间清空
思路:线段树的区间合并。
用cov记录区段的状态,-1代表没有被更新,0代表空闲,1代表是有人入住的。
用lmax代表从左端点开始最长的空闲区间,rmax代表从右开始最长的区间。tree代表自己这个区间内拥有的最大区间。
接下来就是两个重要的函数:pushdown pushup了
void pushup(int num,int mid)
{
lmax[num]=lmax[num<<1];
rmax[num]=rmax[num<<1|1];
if(lmax[num]==(mid-(mid>>1)))lmax[num]+=lmax[num<<1|1];
if(rmax[num]==(mid>>1))rmax[num]+=rmax[num<<1];
tree[num]=max(tree[num<<1],max(tree[num<<1|1],rmax[num<<1]+lmax[num<<1|1]));
}
首先在不考虑左右最大区间重叠的情况下,左边最大区间是左儿子的最大左区间,右边的最大区间是又儿子的最大右区间。
当lmax[num]==(mid-(mid>>1))时,就代表左边全部是空闲的,就要考虑一个最大的区间被分成了两边了,就再加上右儿子的最大左区间。
同理。
最后更新最大区间的时候 就要三者做比较 最大的左 最大的右 左右之间的部分。
#include <iostream>
#include <cstdio>
#include <algorithm>
#define MAXN 50005
using namespace std;int tree[MAXN<<2];
int lmax[MAXN<<2];
int rmax[MAXN<<2];
int cov[MAXN<<2];void pushup(int num,int mid)
{lmax[num]=lmax[num<<1];rmax[num]=rmax[num<<1|1];if(lmax[num]==(mid-(mid>>1)))lmax[num]+=lmax[num<<1|1];if(rmax[num]==(mid>>1))rmax[num]+=rmax[num<<1];tree[num]=max(tree[num<<1],max(tree[num<<1|1],rmax[num<<1]+lmax[num<<1|1]));
}void pushdown(int num,int mid)
{if(cov[num]!=-1){cov[num<<1] = cov[num<<1|1] = cov[num];tree[num<<1] = lmax[num<<1] = rmax[num<<1] = cov[num] ? 0: (mid - (mid>>1));tree[num<<1|1] = lmax[num<<1|1] = rmax[num<<1|1] = cov[num] ? 0:(mid>>1);cov[num]=-1;}
}void build(int num,int l,int r)
{lmax[num] = rmax[num] = tree[num] = r - l + 1;cov[num] = -1;if(r==l)return;int mid = (r+l)>>1;build(num<<1,l,mid);build(num<<1|1,mid+1,r);
}int query(int num,int s,int e,int w)
{if(e==s)return s;pushdown(num,e-s+1);int mid = (s+e)>>1;if(tree[num<<1]>=w)return query(num<<1,s,mid,w);else if(rmax[num<<1] + lmax[num<<1|1] >=w)return mid + 1 - rmax[num<<1];return query(num<<1|1,mid+1,e,w);
}void update(int num,int s,int e,int l,int r,int v)
{if(l<=s && e<=r){tree[num] = lmax[num] = rmax[num] = v ? 0 : e-s+1;cov[num] = v;return ;}pushdown(num,e-s+1);int mid = (e+s)>>1;if(l<=mid)update(num<<1,s,mid,l,r,v);if(r>mid)update(num<<1|1,mid+1,e,l,r,v);pushup(num,e-s+1);
}int main()
{int n,m;scanf("%d%d",&n,&m);build(1,1,n);for(int i=1;i<=m;i++){int a,b,c;scanf("%d",&a);if(a==1){scanf("%d",&b);if(tree[1]<b)puts("0");else {int p=query(1,1,n,b);printf("%d\n",p);update(1,1,n,p,p+b-1,1);}}else{scanf("%d%d",&b,&c);update(1,1,n,b,b+c-1,0);}}return 0;
}