您的位置:首页 > 其它

POJ3667--Hotel(区间合并)

2015-08-17 23:02 417 查看
题目大意:给出一个区间,可以进行两种操作。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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: