poj Hotel 线段树

经典线段树的题。

每个节点存储的信息：左端点连续空房间的长度，右端点连续空房间长度，连续空房间的最大长度。

由于要求每次必须从尽量靠左边的位置进行居住，那么搜索时应尽量让区间起始位置更小：

1.如果当前区间的最大空房间长度小于要寻找的长度，说明不会找到符合的区间，直接退出。

2.如果左子区间的最大空房间长度大于等于要寻找的长度，那么应该进入左子区间查找。

3.否则，如果左子区间的右端点连续空房间长度 + 右子区间的左端点连续空房间的长度 大于等于
要寻找的长度则直接返回左子区间右端点起始空房间下标。

4.否则，进入右区间寻找。

AC代码：

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 2e5;
struct node{
	int l, r;
	int len, lenl, lenr;
	int tag; //0代表全空  1代表全满 -1代表混杂
}t[maxn];

void build(int l, int r, int cur){
	t[cur].l = l, t[cur].r = r;
	t[cur].len = t[cur].lenl = t[cur].lenr = r - l + 1;
	t[cur].tag = 0;
	if(l == r) return;
	int mid = (l + r) / 2;
	build(l, mid, cur << 1);
	build(mid + 1, r, (cur << 1) + 1);
}

void lazy(int cur){
	if(t[cur].l == t[cur].r) return;
	if(t[cur].tag >= 0) {  //覆盖
		int l = cur << 1, r = (cur << 1) + 1;
		int val;
		if(t[cur].tag == 0) val = t[l].r - t[l].l + 1;
		else val = 0;
		t[l].len = t[l].lenl = t[l].lenr = val;

		if(t[cur].tag == 0) val = t[r].r - t[r].l + 1;
		else val = 0;
		t[r].len = t[r].lenl = t[r].lenr = val;
		t[r].tag = t[l].tag = t[cur].tag;
	}
}

int search(int len, int cur){
	if(t[cur].len < len) return 0;
	int l = t[cur].l, r = t[cur].r;
	if(l == r) return l;
	lazy(cur);
	if(t[cur << 1].len >= len) return search(len, cur << 1);
	else if(t[cur << 1].lenr + t[(cur << 1) + 1].lenl >= len) return t[cur << 1].r - t[cur << 1].lenr + 1;
	else if(t[(cur << 1) + 1].len >= len) return search(len, (cur << 1) +1);
}

void update(int l, int r, int tag, int cur){
	int l1 = t[cur].l, r1 = t[cur].r;
	if(l1 == l && r1 == r){
		t[cur].tag = tag;
		int val;
		if(tag) val = 0;
		else val = r1 - l1 + 1;
		t[cur].len = t[cur].lenl = t[cur].lenr = val;
		return;
	}
	lazy(cur);
	int mid = (l1 + r1) / 2;
	if(r <= mid) update(l, r, tag, cur << 1);
	else if(l >= mid + 1) update(l, r, tag, (cur << 1) + 1);
	else {
		update(l, mid, tag, cur << 1);
		update(mid + 1, r, tag, (cur << 1) + 1);
	}

	int c1 = cur << 1, c2 = (cur << 1) + 1;
	int w = t[c1].lenr + t[c2].lenl;
	t[cur].lenl = t[c1].lenl, t[cur].lenr = t[c2].lenr;
	if(t[c1].tag == 0) t[cur].lenl = w;
	if(t[c2].tag == 0) t[cur].lenr = w;
	t[cur].len = max(t[c1].len, t[c2].len);
	t[cur].len = max(w, t[cur].len);
	t[cur].len = max(t[cur].len, t[cur].lenr);
	t[cur].len = max(t[cur].len, t[cur].lenl);
	if(t[cur].len == 0) t[cur].tag = 1;
	else if(t[cur].len == r1 - l1 + 1) t[cur].tag = 0;
	else t[cur].tag = -1;
}

int main(){
	int n,q;
	while(scanf("%d%d", &n, &q) == 2){
		build(1, n, 1);
		int a, b, c;
		for(int i = 0; i < q; ++i){
			scanf("%d", &a);
			if(a == 1){
				scanf("%d", &b);
				int x = search(b, 1);
				printf("%d\n", x);
				if(x != 0) update(x, x + b - 1, 1, 1);
			}
			else {
				scanf("%d%d", &b, &c);
				update(b, b + c - 1, 0, 1);
			}
		}
	}
	return 0;
}

如有不当之处欢迎指出！