传送门:点击打开链接
题目大意:
有N个房间排在一列,有两种操作。
1:查询最靠左的长度为len的空房间,并且入住这些空房间。
2:以l开头,长度为r的房间退房。(如果本来就是空的 还是要退房)。
解题思路:
一类经典的线段树题目。区间合并类。
容易想到在查询上做做手脚,这题就差不多了。问题在于维护哪些东西。由于两个子区间要求合并,那么可以记录一个前最长可用连续,后最长可用连续,和连续可用的最大值。
那么pushup函数和pushdown函数就很简单了。
下面就是查询:如果左区间的连续值是大于等于要求的话,那么查左区间,如果右区间的连续值大于等于要求的话 ,查询右区间。
如果左区间的右连续+右区间的左连续大于等于要求的话。可以直接返回区间的最左下标。
注意:因为要求最偏左,所以注意顺序。
学了几天线段树了,感觉还是没怎么入门。慢慢来吧。
#include <cstdio>
#include <algorithm>
#define maxn 50010
using namespace std;
struct Node
{
int l,r;
int sum,pre,suf;//最长可用区间,左最长,右最长
int c;//0表示未被占用 ,1表示被占用
}tree[maxn<<2];
inline void pushup(int id)
{
if(tree[id<<1].sum == tree[id<<1].r-tree[id<<1].l+1)
tree[id].pre = tree[id<<1].sum + tree[id<<1|1].pre;
else
tree[id].pre = tree[id<<1].pre;
if(tree[id<<1|1].sum == tree[id<<1|1].r-tree[id<<1|1].l+1)
tree[id].suf = tree[id<<1|1].sum + tree[id<<1].suf;
else
tree[id].suf = tree[id<<1|1].suf;
tree[id].sum = max(max(tree[id<<1].sum,tree[id<<1|1].sum),tree[id<<1].suf+tree[id<<1|1].pre);
}
inline void pushdown(int id)
{
if(tree[id].c+1)
{
tree[id<<1].c = tree[id<<1|1].c = tree[id].c;
int len = (tree[id].r- tree[id].l + 1);
int len1 = len-(len>>1);
int len2 = len>>1;
if(tree[id].c)//标记是1,全部标记为不可用
{
tree[id<<1].pre = 0;
tree[id<<1].suf = 0;
tree[id<<1].sum = 0;
tree[id<<1|1].sum = 0;
tree[id<<1|1].pre = 0;
tree[id<<1|1].suf = 0;
}
else//标记是0,表示全部都可用
{
tree[id<<1].pre = len1;
tree[id<<1].suf = len1;
tree[id<<1].sum = len1;
tree[id<<1|1].sum = len2;
tree[id<<1|1].pre = len2;
tree[id<<1|1].suf = len2;
}
tree[id].c = -1;
}
}
void build(int id,int l,int r)
{
tree[id].l = l;
tree[id].r = r;
tree[id].c = -1;
if(l == r)
{
tree[id].sum = 1;
tree[id].suf = 1;
tree[id].pre = 1;
return ;
}
int mid = (l+r)>>1;
build(id<<1,l,mid);
build(id<<1|1,mid+1,r);
pushup(id);
}
void op(int id,int l,int r,int c)
{
if(tree[id].l >= l &&tree[id].r <= r)
{
if(c)
tree[id].pre = tree[id].suf = tree[id].sum = 0;
else
tree[id].pre = tree[id].suf = tree[id].sum = tree[id].r-tree[id].l+1;
tree[id].c = c;
return ;
}
pushdown(id);
int mid = (tree[id].l + tree[id].r) >>1;
if(l <= mid ) op(id<<1,l,r,c);
if(mid < r) op(id<<1|1,l,r,c);
pushup(id);
}
int query(int id,int len)
{
if(tree[id].l == tree[id].r) return tree[id].l;
pushdown(id);
int mid = (tree[id].l + tree[id].r)>>1;
if(tree[id<<1].sum >= len)
return query(id<<1,len);
else if(tree[id<<1].suf + tree[id<<1|1].pre >= len)
return mid - tree[id<<1].suf+1;
else
return query(id<<1|1,len);
}
int main()
{
int n,q;
while(scanf("%d %d",&n,&q) != EOF)
{
build(1,1,n);
while(q--)
{
int ch;
scanf("%d",&ch);
if(ch == 1)
{
int len;
scanf("%d",&len);
if(tree[1].sum < len)
{
printf("0\n");
continue;
}
int pos= query(1,len);
printf("%d\n",pos);
op(1,pos,pos+len-1,1);
}
else
{
int l,r;
scanf("%d %d\n",&l,&r);
op(1,l,l+r-1,0);
}
}
}
return 0;
}