首页 > 学院 > 开发设计 > 正文

poj Hotel 线段树

2019-11-11 02:45:46
字体:
来源:转载
供稿:网友

      经典线段树的题。

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

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

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;}如有不当之处欢迎指出!


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表