[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，感觉很爽。

1 #include <algorithm>
2 #include <iostream>
3 #include <iomanip>
4 #include <cstring>
5 #include <climits>
6 #include <complex>
7 #include <cassert>
8 #include <cstdio>
9 #include <bitset>
10 #include <vector>
11 #include <deque>
12 #include <queue>
13 #include <stack>
14 #include <ctime>
15 #include <set>
16 #include <map>
17 #include <cmath>
18 using namespace std;
19
20 #define lrt rt << 1
21 #define rrt rt << 1 | 1
22 typedef struct Seg { int sign, ls, ms, rs; }Seg;
23 const int maxn = 50500;
24 Seg seg[maxn<<2];
25 int val[maxn];
26 int n, m;
27
28 void pushup(int l, int r, int rt) {
29     int mid = (l + r) >> 1;
30     int ll = mid - l + 1, rr = r - mid;
31     seg[rt].ls = seg[lrt].ls, seg[rt].rs = seg[rrt].rs;
32     if(seg[lrt].rs==ll) seg[rt].ls += seg[rrt].ls;
33     if(seg[rrt].ls==rr) seg[rt].rs += seg[lrt].rs;
34     seg[rt].ms = max(seg[lrt].ms, seg[rrt].ms);
35     seg[rt].ms = max(seg[rt].ms, seg[lrt].rs+seg[rrt].ls);
36 }
37
38 void pushdown(int l, int r, int rt) {
39     int mid = (l + r) >> 1;
40     if(seg[rt].sign == -1) return;
41     if(seg[rt].sign == 1) {
42         seg[lrt].sign = seg[rrt].sign = 1;
43         seg[lrt].ms = seg[rrt].ms = 0;
44         seg[lrt].ls = seg[rrt].ls = 0;
45         seg[lrt].rs = seg[rrt].rs = 0;
46     }
47     else {
48         seg[lrt].sign = seg[rrt].sign = 0;
49         seg[lrt].ls = seg[lrt].ms = seg[lrt].rs = mid - l + 1;
50         seg[rrt].ls = seg[rrt].ms = seg[rrt].rs = r - mid;
51     }
52     seg[rt].sign = -1;
53 }
54
55 void build(int l, int r, int rt) {
56     if(l == r) {
57         seg[rt].sign = -1;
58         seg[rt].ls = seg[rt].ms = seg[rt].rs = r - l + 1;
59         return;
60     }
61     int mid = (l + r) >> 1;
62     build(l, mid, lrt);
63     build(mid+1, r, rrt);
64     pushup(l, r, rt);
65 }
66
67 void update(int L, int R, int sign, int l, int r, int rt) {
68     if(L <= l && r <= R) {
69         if(sign) seg[rt].ls = seg[rt].ms = seg[rt].rs = 0;
70         else seg[rt].ls = seg[rt].ms = seg[rt].rs = r - l + 1;
71         seg[rt].sign = sign;
72         return;
73     }
74     pushdown(l, r, rt);
75     int mid = (l + r) >> 1;
76     if(L <= mid) update(L, R, sign, l, mid, lrt);
77     if(mid < R) update(L, R, sign, mid+1, r, rrt);
78     pushup(l, r, rt);
79 }
80
81 int query(int len, int l, int r, int rt) {
82     if(l == r) return rt;
83     pushdown(l, r, rt);
84     int mid = (l + r) >> 1;
85     if(seg[lrt].ms >= len) return query(len, l, mid, lrt);
86     else if(seg[lrt].rs + seg[rrt].ls >= len) return mid - seg[lrt].rs + 1;
87     else return query(len, mid+1, r, rrt);
88 }
89
90 int main() {
91     // freopen("in", "r", stdin);
92     int x, l, r;
93     while(~scanf("%d%d",&n,&m)) {
94         memset(val, 0, sizeof(val));
95         build(1, n, 1);
96         while(m--) {
97             scanf("%d",&x);
98             if(x == 1) {
99                 scanf("%d",&l);
100                 if(l > seg[1].ms) {
101                     puts("0");
102                     continue;
103                 }
104                 r = query(l, 1, n, 1);
105                 printf("%d\n", r);
106                 update(r, r+l-1, 1, 1, n, 1);
107             }
108             else {
109                 scanf("%d%d",&l,&r);
110                 update(l, l+r-1, 0, 1, n, 1);
111             }
112         }
113     }
114     return 0;
115 }