poj 3667 Hotel(线段树中级,区间合并)

来源:互联网 发布:js常用正则表达式 编辑:程序博客网 时间:2024/06/07 15:13

题意:
。。。
思路:
类似动态的最大连续和。
这里我们需要构建一个查询 最靠左的长度不小于D的区间的操作。
可以分类讨论 lc = left child, rc = right child
0)max[o]<D
不存在,直接返回
1)lmax[lc]>=D
左半区间查询
2)rmax[lc]+lmax[rc]>=D,即跨越中点
返回 midlmax[rc]+1
3)rmax[lc]>=D
在右半区间查询

#define lson(x) ((x)<<1)#define rson(x) (((x)<<1)|1)struct node {    int l, r, m, tag;};node a[Maxn*4+5];void upd(int o, int len, int flag) {    a[o].l = a[o].r = a[o].m = (flag ? 0 : len);    a[o].tag = flag;}void pd(int L, int R, int o) {    if (a[o].tag != -1) {       int lc = lson(o), rc = rson(o), mid = (L+R)>>1;       upd(lc, mid-L+1, a[o].tag);       upd(rc, R-mid, a[o].tag);       a[o].tag = -1;    }}void pu(int L, int R, int o) {    int lc = lson(o), rc = rson(o), mid = (L+R)>>1;    int l1 = mid - L + 1, l2 = R - mid;    a[o].l = a[lc].l;a[o].r = a[rc].r;    if (a[lc].l == l1) a[o].l += a[rc].l;    if (a[rc].l == l2) a[o].r += a[lc].r;    a[o].m = max ( a[lc].r + a[rc].l, max (a[lc].m, a[rc].m)  );}void seg_update(int L, int R, int o, int qL, int qR, int v) {    if (qL <= L && R <= qR) {        upd(o, R-L+1, v);return;    }    pd(L, R, o);    int lc = lson(o), rc = rson(o), mid = (L+R) >> 1;    if (qL <= mid) seg_update(L, mid, lc, qL, qR, v);    if (qR > mid) seg_update(mid+1, R, rc, qL, qR, v);    pu(L, R, o);}int seg_query(int L, int R, int o, int x) {    if (a[o].m < x) return -1;    int lc = lson(o), rc = rson(o), mid = (L+R) >> 1;    if (a[lc].m >= x) return seg_query(L, mid, lc, x);    if (a[lc].r + a[rc].l >= x) return mid - a[lc].r + 1;    return seg_query(mid+1, R, rc, x);}void seg_build(int L, int R, int o) {    int len = R-L+1;a[o] = (node) {len, len, len, -1};    if (L < R) {        int mid = (L+R) >> 1;        seg_build(L, mid, lson(o));seg_build(mid+1, R, rson(o));    }}int main(){#ifndef ONLINE_JUDGE    freopen("input.in", "r", stdin);#endif    int x, y, z;    while (scanf("%d%d", &n, &m) != EOF) {        seg_build(1, n, 1);        rep(i, 1, m) {            scanf("%d", &x);            if (x == 1) {                scanf("%d", &y);                int s = seg_query(1, n, 1, y);                printf("%d\n", s == -1 ? 0 : s);                if (s != -1) {                    seg_update(1, n, 1, s, s+y-1, 1);                }                //printf("check in %d: %d\n", y, s);            } else {                scanf("%d%d", &y, &z);                seg_update(1, n, 1, y, y+z-1, 0);                //printf("check out: %d %d\n", y, z);            }        }    }    return 0;}
0 0
原创粉丝点击