经典线段树的题。
每个节点存储的信息:左端点连续空房间的长度,右端点连续空房间长度,连续空房间的最大长度。
由于要求每次必须从尽量靠左边的位置进行居住,那么搜索时应尽量让区间起始位置更小:
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;}如有不当之处欢迎指出!
新闻热点
疑难解答