【线段树】POJ 3667 Hotel 区间合并

水题好像来的晚了一些

题意:

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

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

在最底层用1表示空，0表示占有这房间。。

sum[rt]表示该节点下最大连续空房间数

lsum[rt]表示该节点从左端开始连续的空房间数

rsum[rt]表示该节点从右端开始连续的空房间数

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#include <queue>
#include <stack>
#include <vector>
#include <deque>
#include <set>
#include <map>
#define IN freopen ("in.txt" , "r" , stdin);
#define OUT freopen ("out.txt" , "w" , stdout);
typedef long long LL;
const int MAXN = 100110;//点数的最大值
const int MAXM = 20006;//边数的最大值
const int INF = 11521204;
const int mod=1000000007;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int sum[MAXN<<2],lsum[MAXN<<2],rsum[MAXN<<2];
int col[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];
sum[rt]=max(max(sum[rt<<1],sum[rt<<1|1]),rsum[rt<<1]+lsum[rt<<1|1]);
}
void pushdown(int rt,int m)
{
if(col[rt]!=-1)
{
col[rt<<1]=col[rt<<1|1]=col[rt];
sum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=col[rt]?0:m-(m>>1);
sum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=col[rt]?0:m>>1;
col[rt]=-1;
}
}
void build(int l,int r,int rt)
{
sum[rt]=lsum[rt]=rsum[rt]=r-l+1;
col[rt]=-1;
if(l==r)
return ;
int m=(l+r)>>1;
build(lson);
build(rson);
}
void update(int L,int R,int cc,int l,int r,int rt)
{
if(L<=l&&r<=R)
{
sum[rt]=lsum[rt]=rsum[rt]=cc? 0:r-l+1;
col[rt]=cc;
return ;
}
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(L<=m) update(L,R,cc,lson);
if(m<R) update(L,R,cc,rson);
pushup(rt,r-l+1);
}
int query(int x,int l,int r,int rt)
{
if(l==r)
return l;
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(sum[rt<<1]>=x)
return query(x,lson);
else if(rsum[rt<<1]+lsum[rt<<1|1]>=x)
return m-rsum[rt<<1]+1;
else return query(x,rson);
}
int main()
{
int n,m;
//IN;
while(scanf("%d%d",&n,&m)!=EOF)
{
build(1,n,1);
while(m--)
{
int op,k,d;
scanf("%d",&op);
if(op==1)
{
scanf("%d",&d);
if(sum[1]<d)
printf("0\n");
else
{
int a=query(d,1,n,1);
printf("%d\n",a);
update(a,a+d-1,1,1,n,1);
}
}
else
{
scanf("%d%d",&k,&d);
update(k,k+d-1,0,1,n,1);
}
}
}
return 0;
}