POJ 3667 Hotel（区间合并）

题目地址：http://poj.org/problem?id=3667

题意：1 a:询问是不是有连续长度为a的空房间,有的话住进最左边

2 a b:将[a,a+b-1]的房间清空

思路:记录区间中最长的空房间

思路：update区间替换，query求满足条件的最左边的一个房间

AC代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cstring>
#include <climits>
#include <cmath>
#include <cctype>
const int inf = 0x3f3f3f3f;//1061109567
typedef long long LL;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn = 50010;
using namespace std;
int lsum[maxn<<2],rsum[maxn<<2],msum[maxn<<2],cover[maxn<<2];
void pushdown(int rt,int m)
{
if(cover[rt] != -1)
{
lsum[rt<<1]=rsum[rt<<1]=msum[rt<<1] = cover[rt]?0:(m-(m>>1));
lsum[rt<<1|1]=rsum[rt<<1|1]=msum[rt<<1|1]=cover[rt]?0:(m>>1);
cover[rt<<1]=cover[rt<<1|1]=cover[rt];
cover[rt]=-1;
}
}
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(rsum[rt<<1]+lsum[rt<<1|1],max(msum[rt<<1],msum[rt<<1|1]));
}
void build(int l,int r,int rt)
{
lsum[rt]=rsum[rt]=msum[rt]=r-l+1;
cover[rt] = -1;
if(l == r) return;
int m = (l+r)>>1;
build(lson);
build(rson);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if(l >= L && r <= R)
{
lsum[rt]=rsum[rt]=msum[rt]=c?0:r-l+1;
cover[rt] = c;
return;
}
pushdown(rt,r-l+1);
int m = (l + r)>>1;
if(L <= m) update(L,R,c,lson);
if(R > m) update(L,R,c,rson);
pushup(rt,r-l+1);
}

int query(int c,int l,int r,int rt)
{
if(l == r) return l;
pushdown(rt,r-l+1);
int m = (l + r)>>1;
if(msum[rt<<1]>=c) return query(c,lson);
else if(rsum[rt<<1]+lsum[rt<<1|1] >=c) return m - rsum[rt<<1]+1;
return query(c,rson);
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m) != EOF)
{
build(1,n,1);
int op,a,b;
while(m--)
{
scanf("%d",&op);
if(op == 1)
{
scanf("%d",&a);
if(msum[1]<a) puts("0");
else
{
int q = query(a,1,n,1);
printf("%d\n",q);
update(q,q+a-1,1,1,n,1);
}
}
else
{
scanf("%d%d",&a,&b);
update(a,a+b-1,0,1,n,1);
}
}
}
return 0;
}