POJ3667--Hotel(区间合并)

题目大意:给出一个区间,可以进行两种操作。1、查询是否有长度为X的连续区间,有的话,输出左端点。2、清空某段连续区间


分析:线段树,区间合并。

sum[i]表示当前最长连续区间的长度

lsum[i]表示从当前区间左端点起,最大连续区间的长度

rsum[i]表示从当前区间右端点起,最大连续区间的长度

covered[i]表示是否住了人,0表示没有人,1表示有人,-1表示不需要更新。当下次更新或者查询时,若为0或1,则要向下更新上面三个数组,更新完后当前covered赋值为-1。这个数组就体现了延迟更新(懒惰标记)

接着,就是关于上面三个数组的维护。

当lsum[i]等于左半个区间长度,则lsum[i]加上右半边区间的lsum,也就是lsum[i] += lsum[2*i+2]。

当rsum[i]等于右半个区间长度,则rsum[i]加上左半边区间的rsum,也就是rsum[i] += rsum[2*i+1]。

sum[i] = max{ rsum[2*i+1]+lsum[2*i+2],sum[2*i+1],sum[2*i+2] }。

这里便体现了区间合并。


代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 50010;
int n, m;
int sum[4*maxn], lsum[4*maxn], rsum[4*maxn];
int covered[4*maxn];

void PushUp(int root, int len) {
    lsum[root] = lsum[2*root+1];
    rsum[root] = rsum[2*root+2];
    if(lsum[root] == len-len/2)
        lsum[root] += lsum[2*root+2];
    if(rsum[root] == len/2)
        rsum[root] += rsum[2*root+1];
    sum[root] = max(rsum[2*root+1]+lsum[2*root+2], max(sum[2*root+1], sum[2*root+2]));

}

void PushDown(int root, int len) {
    if(covered[root] != -1) {
        covered[2*root+1] = covered[2*root+2] = covered[root];
        sum[2*root+1] = lsum[2*root+1] = rsum[2*root+1] = covered[root] ? 0 : (len-len/2);
        sum[2*root+2] = lsum[2*root+2] = rsum[2*root+2] = covered[root] ? 0 : (len/2);
        covered[root] = -1;
    }
}

void Build(int root, int l, int r) {
    sum[root] = lsum[root] = rsum[root] = r-l+1;
    covered[root] = -1;
    if(l == r) return;
    Build(2*root+1, l, (l+r)/2);
    Build(2*root+2, (l+r)/2+1, r);
    return;
}

void Update(int root, int l, int r, int L, int R, int c) {
    if(L <= l && r <= R) {
        sum[root] = lsum[root] = rsum[root] = c ? 0 : (r-l+1);
        covered[root] = c;
        return;
    }
    PushDown(root, r-l+1);
    int m = (l+r)/2;
    if(L <= m)
        Update(2*root+1, l, m, L, R, c);
    if(R > m)
        Update(2*root+2, m+1, r, L, R, c);
    PushUp(root, r-l+1);
    return;
}

int Query(int root, int l, int r, int x) {
    if(l == r) return l;
    PushDown(root, r-l+1);
    int m = (l+r)/2;
    if(sum[2*root+1] >= x)
        return Query(2*root+1, l, m, x);
    else if(rsum[2*root+1]+lsum[2*root+2] >= x) 横跨左右孩子且连续的区间可以满足,那么可以直接返回下标
        return m-rsum[2*root+1]+1;
    else
        return Query(2*root+2, m+1, r, x);
}



int main() {
    scanf("%d%d", &n, &m);
    Build(0, 1, n);
    while(m--) {
        int op, a, b;
        scanf("%d", &op);
        if(op == 1) {
            scanf("%d", &a);
            if(sum[0] < a) puts("0");
            else {
                int pos = Query(0, 1, n, a);
                printf("%d\n", pos);
                Update(0, 1, n, pos, pos+a-1, 1);
            }
        }
        else {
            scanf("%d%d", &a, &b);
            Update(0, 1, n, a, a+b-1, 0);
        }
    }
    return 0;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值