[POJ3667]Hotel（线段树，区间合并，重写）

 

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

题意：有一个hotel有n间房子，现在有2种操作：

1 a，check in，表示入住。需要a间连续的房子。返回尽量靠左的房间编号并更新。

2 a b，check out，从a开始退房，一共退到a+b-1。

seg数组l表示从左边开始数空房子的个数，r表示从右边数空房子的个数，s表示一共最多是有s个连续的空房子。

add作为延迟标记有3种状态：-1表示不操作，0表示退房，1表示入住。

重写1A，感觉很爽。

#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>
using namespace std;

#define lrt rt << 1
#define rrt rt << 1 | 1
typedef struct Seg { int sign, ls, ms, rs; }Seg;
const int maxn = 50500;
Seg seg[maxn<<2];
int val[maxn];
int n, m;

void pushup(int l, int r, int rt) {
    int mid = (l + r) >> 1;
    int ll = mid - l + 1, rr = r - mid;
    seg[rt].ls = seg[lrt].ls, seg[rt].rs = seg[rrt].rs;
    if(seg[lrt].rs==ll) seg[rt].ls += seg[rrt].ls;
    if(seg[rrt].ls==rr) seg[rt].rs += seg[lrt].rs;
    seg[rt].ms = max(seg[lrt].ms, seg[rrt].ms);
    seg[rt].ms = max(seg[rt].ms, seg[lrt].rs+seg[rrt].ls);
}

void pushdown(int l, int r, int rt) {
    int mid = (l + r) >> 1;
    if(seg[rt].sign == -1) return;
    if(seg[rt].sign == 1) {
        seg[lrt].sign = seg[rrt].sign = 1;
        seg[lrt].ms = seg[rrt].ms = 0;
        seg[lrt].ls = seg[rrt].ls = 0;
        seg[lrt].rs = seg[rrt].rs = 0;
    }
    else {
        seg[lrt].sign = seg[rrt].sign = 0;
        seg[lrt].ls = seg[lrt].ms = seg[lrt].rs = mid - l + 1;
        seg[rrt].ls = seg[rrt].ms = seg[rrt].rs = r - mid;
    }
    seg[rt].sign = -1;
}

void build(int l, int r, int rt) {
    if(l == r) {
        seg[rt].sign = -1;
        seg[rt].ls = seg[rt].ms = seg[rt].rs = r - l + 1;
        return;
    }
    int mid = (l + r) >> 1;
    build(l, mid, lrt);
    build(mid+1, r, rrt);
    pushup(l, r, rt);
}

void update(int L, int R, int sign, int l, int r, int rt) {
    if(L <= l && r <= R) {
        if(sign) seg[rt].ls = seg[rt].ms = seg[rt].rs = 0;
        else seg[rt].ls = seg[rt].ms = seg[rt].rs = r - l + 1;
        seg[rt].sign = sign;
        return;
    }
    pushdown(l, r, rt);
    int mid = (l + r) >> 1;
    if(L <= mid) update(L, R, sign, l, mid, lrt);
    if(mid < R) update(L, R, sign, mid+1, r, rrt);
    pushup(l, r, rt);
}

int query(int len, int l, int r, int rt) {
    if(l == r) return rt;
    pushdown(l, r, rt);
    int mid = (l + r) >> 1;
    if(seg[lrt].ms >= len) return query(len, l, mid, lrt);
    else if(seg[lrt].rs + seg[rrt].ls >= len) return mid - seg[lrt].rs + 1;
    else return query(len, mid+1, r, rrt);
}

int main() {
    // freopen("in", "r", stdin);
    int x, l, r;
    while(~scanf("%d%d",&n,&m)) {
        memset(val, 0, sizeof(val));
        build(1, n, 1);
        while(m--) {
            scanf("%d",&x);
            if(x == 1) {
                scanf("%d",&l);
                if(l > seg[1].ms) {
                    puts("0");
                    continue;
                }
                r = query(l, 1, n, 1);
                printf("%d
", r);
                update(r, r+l-1, 1, 1, n, 1);
            }
            else {
                scanf("%d%d",&l,&r);
                update(l, l+r-1, 0, 1, n, 1);
            }
        }
    }
    return 0;
}