题目大意:给出一个区间,可以进行两种操作。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;
}