POJ 3667 Hotel(区間合併)

タイトルの住所:http://poj.org/problem?id=3667
A:連続長さがaの空き部屋があるかどうか聞いてください.ある場合は一番左の2 a:b:[a,a+b-1]の部屋を空欄にします.記録区間で一番長い空き部屋です.
考え方:uudate区間の交替、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
",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;
}