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);
            }
        }
    }
}