POJ 3667 Hotel（线段树区间合并与查询）
题意：
有N个房间，M次操作。1 a表示找到连续的长度为a的空房间，如果有多解，优先左边的，即表示入住。2 b len把起点为b长度的len的房间清空，即退房。
思路：
线段树的区间合并和查询：
1. sum[]表示区间内最大的连续房间数
2. lsum[], rsum[]分别表示区间从左边、右边起的连续房间数目
3. 主要是其中的更新过程比较繁琐，理清楚逻辑就不是那么难了
#include <cstdio>
#include <algorithm>
using namespace std;
#define lhs l, m, rt << 1
#define rhs m + 1, r, rt << 1 | 1
const int maxn = 50010;
int sum[maxn << 2], lab[maxn << 2];
int lsum[maxn << 2], rsum[maxn << 2];
enum __check { checkout = 0, checkin };
void PushDown(int rt, int m)
{
if (lab[rt] != -1)
{
lab[rt << 1] = lab[rt << 1 | 1] = lab[rt];
sum[rt << 1] = lsum[rt << 1] = rsum[rt << 1] = lab[rt] ? 0 : m - (m >> 1);
sum[rt << 1 | 1] = lsum[rt << 1 | 1] = rsum[rt << 1 | 1] = lab[rt] ? 0 : (m >> 1);
lab[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];
sum[rt] = max(rsum[rt << 1] + lsum[rt << 1 | 1], max(sum[rt << 1], sum[rt << 1 | 1]));
}
void Build(int l, int r, int rt)
{
sum[rt] = lsum[rt] = rsum[rt] = r - l + 1;
lab[rt] = -1;
if (l == r)
return ;
int m = (l + r) >> 1;
Build(lhs);
Build(rhs);
}
void Update(int beg, int end, __check c, int l, int r, int rt)
{
if (beg <= l && r <= end)
{
sum[rt] = lsum[rt] = rsum[rt] = c ? 0 : r - l + 1;
lab[rt] = c;
return ;
}
PushDown(rt, r - l + 1);
int m = (l + r) >> 1;
if (beg <= m)
Update(beg, end, c, lhs);
if (end > m)
Update(beg, end, c, rhs);
PushUp(rt, r - l + 1);
}
int Query(int w, int l, int r, int rt)
{
if (l == r)
return r;
PushDown(rt, r - l + 1);
int m = (l + r) >> 1;
if (sum[rt << 1] >= w)
return Query(w, lhs);
else if (rsum[rt << 1] + lsum[rt << 1 | 1] >= w)
return m - rsum[rt << 1] + 1;
else
return Query(w, rhs);
}
int main()
{
int n, m;
scanf("%d %d", &n, &m);
Build(1, n, 1);
for (int i = 0; i < m; ++i)
{
int op, a, b;
scanf("%d", &op);
if (op == 1)
{
scanf("%d", &a);
if (sum[1] >= a)
{
int p = Query(a, 1, n, 1);
printf("%d\n", p);
Update(p, p + a - 1, checkin, 1, n, 1);
}
else
printf("0\n");
}
else if (op == 2)
{
scanf("%d %d", &a, &b);
Update(a, a + b - 1, checkout, 1, n, 1);
}
}
return 0;
}