POJ 3667 Hotel

题目链接：http://poj.org/problem?id=3667

题意：有n个连续的房间，有2种操作，一种是有若干个客人入住，要选择连续且尽量靠左的房间，输出最靠左的房间若没有则输出0，一种是将清空di后到di+x的房间清空

思路：第一次做区间合并的题目主主要理解一下线段树的3个变量的用意义和更新方法

lsum：从该区间左边界开始有多少个连续的房间，比如：0 0 0 1 1 1 有3个连续的房间，更新时候等于其左子树的lsum，如果整个左子树都是连续的话则要加上右子树的lsum

rsum:从该区间有边界开始有多少个连续的房间，比如:1 1 1 0 1 0有1个连续的房间，更新的时候和lsum同理

tsum:保存的是整个区间最长的连续房间，更新的时候取左子树的rsum加上右子树lsum的和(横跨左右2个区间)，以及左子树的tsum和右子树的tsum就可以了

lazy延迟标记，保存的只有2个值，1和0，1代表该区间以及有住客了，0则是该区间被清空，更新时还是往下推lazy标记然后再向上递归更新

查询的时候尽量往左子树，分为3种步骤：

1.左子树若tsum>=len，则代表左子树有len个连续的房间

2.若左子树的rsum加上右子树的lsum>=len，则跨区间有连续的房间，直接mid-tree[root<<1].rsum+1得出答案

3.若右子树tsum>=len，则代表右子树有len个连续的房间

由于要满足房间尽量靠左所以必须按顺序来进行(询问前要先判断整个区间是否有连续的区间，我在询问时加语句判断莫名wa了，希望有人能指明错误)

代码写的又长又臭……还要改进一下

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define maxn 50030
using namespace std;

struct Tree
{
int l,r;
int lsum,rsum,tsum;;
}tree[maxn*3];

int lazy[maxn*3];
int n,m;

void Pushup(int root)
{
int ll=tree[root<<1].r-tree[root<<1].l+1;
int rr=tree[root<<1|1].r-tree[root<<1|1].l+1;
tree[root].lsum=tree[root<<1].lsum;
tree[root].rsum=tree[root<<1|1].rsum;

if (tree[root<<1].lsum==ll) tree[root].lsum+=tree[root<<1|1].lsum;
if (tree[root<<1|1].rsum==rr) tree[root].rsum+=tree[root<<1].rsum;
int tem=max(tree[root<<1].tsum,tree[root<<1|1].tsum);
tree[root].tsum=max(tree[root<<1].rsum+tree[root<<1|1].lsum,tem);
}

void Pushdown(int root)
{
if (lazy[root]!=-1)
{
int ll=tree[root<<1].r-tree[root<<1].l+1;
int rr=tree[root<<1|1].r-tree[root<<1|1].l+1;
lazy[root<<1]=lazy[root];
lazy[root<<1|1]=lazy[root];
if (lazy[root]==0)
{
tree[root<<1].lsum=ll;
tree[root<<1].rsum=ll;
tree[root<<1].tsum=ll;
tree[root<<1|1].lsum=rr;
tree[root<<1|1].rsum=rr;
tree[root<<1|1].tsum=rr;
}
else
{
tree[root<<1].lsum=0;
tree[root<<1].rsum=0;
tree[root<<1].tsum=0;
tree[root<<1|1].lsum=0;
tree[root<<1|1].rsum=0;
tree[root<<1|1].tsum=0;
}

lazy[root]=-1;
}
}

void build(int root,int l,int r)
{
tree[root].l=l;
tree[root].r=r;
tree[root].lsum=r-l+1;
tree[root].rsum=r-l+1;
tree[root].tsum=r-l+1;
if (l==r) return;

int mid=(l+r)>>1;
build(root<<1,l,mid);
build(root<<1|1,mid+1,r);
Pushup(root);
}

void update(int root,int l,int r,int val)
{
if (tree[root].l>=l && tree[root].r<=r)
{
if (val==1)
{
tree[root].lsum=0;
tree[root].rsum=0;
tree[root].tsum=0;
}
else
{
tree[root].lsum=tree[root].r-tree[root].l+1;
tree[root].rsum=tree[root].r-tree[root].l+1;
tree[root].tsum=tree[root].r-tree[root].r+1;
}

lazy[root]=val;
return;
}
Pushdown(root);
int mid=(tree[root].l+tree[root].r)>>1;
if (l<=mid) update(root<<1,l,r,val);
if (r>mid) update(root<<1|1,l,r,val);
Pushup(root);
}

int que(int root,int len)
{
if (tree[root].l==tree[root].r) return tree[root].l;
Pushdown(root);
int mid=(tree[root].r+tree[root].l)>>1;
if (len<=tree[root<<1].tsum) return que(root<<1,len);
else if (len<=tree[root<<1].rsum+tree[root<<1|1].lsum) return mid-tree[root<<1].rsum+1;
else if (len<=tree[root<<1|1].tsum) return que(root<<1|1,len);
//else return 0; 在这里判断不知道为什么不行……
}

int main()
{

while (~scanf("%d%d",&n,&m))
{
memset(lazy,-1,sizeof(lazy));
build(1,1,n);
for (int i=0;i<m;i++)
{
int op;
scanf("%d",&op);
if (op==1)
{
int a;
scanf("%d",&a);
if(tree[1].tsum < a) printf("0\n");//这里没有判断，在que里判断不知道为什么错了
else
{
int b = que(1,a);
printf("%d\n",b);
update(1,b,b+a-1,1);
}
}
else
{
int a,b;
scanf("%d%d",&a,&b);
update(1,a,a+b-1,0);
}
}
}
}