题目
有n个房间,m次操作,操作 1 x表示有x个人要入住,并且房间要挨着,就是要找到连续的x个房间,输出左端点值,操作 2 x ,y表示x~x+y-1的房间清空了。
又是日常不会然后去百度,线段树花样也太多了吧。
维护三个值sum,lmax,rmax,分别表示区间最大连续,从左端点开始的最大连续,和从右端点开始的最大连续。
t
r
e
e
[
n
o
d
e
]
.
s
u
m
显
然
是
M
a
x
(
t
r
e
e
[
n
o
d
e
<
<
1
]
.
s
u
m
,
t
r
e
e
[
n
o
d
e
<
<
1
∣
1
]
.
s
u
m
,
t
r
e
e
[
n
o
d
e
<
<
1
]
.
r
m
a
x
+
t
r
e
e
[
n
o
d
e
<
<
1
∣
1
]
.
l
m
a
x
)
.
tree[node].sum显然是Max(tree[node<<1].sum, tree[node<<1|1].sum,tree[node<<1].rmax+tree[node<<1|1].lmax).
tree[node].sum显然是Max(tree[node<<1].sum,tree[node<<1∣1].sum,tree[node<<1].rmax+tree[node<<1∣1].lmax).因为区间的最大连续要么在左区间要么再右区间,要么在左右区间合并后的中间部分。确定关系之后就可以开始写了。
代码
#include <iostream>
#include <cstring>
#include <cmath>
#include <bitset>
#include <queue>
#include <vector>
#include <cstdio>
#include <set>
#include <sstream>
#include <string>
#include <algorithm>
#include <map>
#define rep(i, a, b) for (int i = a; i <= b; i++)
#define frep(i, a, b) for (int i = a; i >= b; i--)
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a*b/gcd(a,b)
using namespace std;
const int N = 1e7 + 7;
const int M = 1e2+7;
const int inf = 0x3f3f3f3f;
const int mod = 1e9+7;
typedef long long ll;
int test_prime[8]={2,3,5,7,11,13,17,19};
struct node
{
int sum,lmax,rmax,lazy;
}tree[N<<2];
void build(int node,int l,int r)
{
tree[node].sum=tree[node].lmax=tree[node].rmax=r-l+1;
tree[node].lazy=-1;
if(l==r)return;
int mid=(l+r)>>1;
build(node<<1,l,mid);
build(node<<1|1,mid+1,r);
}
void pushup(int node,int l,int r)
{
int mid=(l+r)>>1;
tree[node].lmax=tree[node<<1].lmax;
if(tree[node<<1].sum==mid-l+1)tree[node].lmax=tree[node<<1].sum+tree[node<<1|1].lmax;
//左区间全部连续,所以可以和右区间的左区间合并
tree[node].rmax=tree[node<<1|1].rmax;
if(tree[node<<1|1].sum==r-mid)tree[node].rmax=tree[node<<1].rmax+tree[node<<1|1].sum;//同上
tree[node].sum=max(tree[node<<1].rmax+tree[node<<1|1].lmax,max(tree[node<<1].sum,tree[node<<1|1].sum));
}
void pushdown(int node,int l,int r)
{
int mid=(l+r)>>1;
tree[node<<1].lazy=tree[node<<1|1].lazy=tree[node].lazy;
tree[node<<1].sum=tree[node<<1].lmax=tree[node<<1].rmax=(mid-l+1)*tree[node].lazy;
//lazy取0,1分别表示入住和清空
tree[node<<1|1].sum=tree[node<<1|1].lmax=tree[node<<1|1].rmax=(r-mid)*tree[node].lazy;
tree[node].lazy=-1;
}
void update(int node,int L,int R,int l,int r,int s)
{
if(L<=l&&r<=R){
tree[node].lazy=s;
tree[node].sum=tree[node].lmax=tree[node].rmax=(r-l+1)*s;
return;
}
if(tree[node].lazy!=-1){
pushdown(node,l,r);
}
int mid=(l+r)>>1;
if(L<=mid)update(node<<1,L,R,l,mid,s);
if(R>mid)update(node<<1|1,L,R,mid+1,r,s);
pushup(node,l,r);
}
int query(int node,int val,int l,int r)
{
if(tree[node].lazy!=-1){
pushdown(node,l,r);
}
if(l==r)return l;
int mid=(l+r)>>1;
if(tree[node<<1].sum>=val)return query(node<<1,val,l,mid);
else if(tree[node<<1].rmax+tree[node<<1|1].lmax>=val){
return mid-tree[node<<1].rmax+1;
//rmax+lmax>=val,说明中间部分可以。因为mid是左区间的右端点,所以开始位置就是mid-rmax+1.
}
else return query(node<<1|1,val,mid+1,r);
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
build(1,1,n);
rep(i,1,m){
int op;
scanf("%d",&op);
if(op&1){
int v;
scanf("%d",&v);
if(v>tree[1].sum){
puts("0");
continue;
}
int pos=query(1,v,1,n);
printf("%d\n",pos);
update(1,pos,pos+v-1,1,n,0);
}else{
int u,v;
scanf("%d%d",&u,&v);
update(1,u,u+v-1,1,n,1);
}
}
return 0;
}