Solution
第二道线段树区间合并的题
第一道稍微简单一点的线段树区间合并题戳这
细节看上一篇吧
一下是需要注意的地方
(来自poj讨论区)
假设max_len表示最大可用连续区间,start_left_len是左起最大可用连续区间,end_right_len是结束于right的最大可用连续区间,预定房间数为bookNum。则有:
1、如果1~n区间的max_len值都比bookNum小,就无解,否则有解。
2、如果start_left_len>=bookNum,返回left
3、如果它的左儿子的max_len值大于等于bookNum,到左儿子里去找。
4、如果左儿子的end_right_len加上右儿子的start_left_len大于等于bookNum,直接返回mid-leftChild.end_right_len+1,一定记得是直接返回,不需要再递归查询,我在这里WA了
5、否则到右儿子里去找。
上面每一步只有前面的每一步都不成立时,才能执行到下一步
Code
update和cl两个函数可以写成一个
// by spli
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
const int N=50010;
int n,m;
struct node{
int L,R,len;
int ms,ls,rs;
int tag;
}t[N*4+5];
int loc;
void pushup(int pos){
t[pos].ls=t[pos<<1].ls;
if(t[pos<<1].ms==t[pos<<1].len) t[pos].ls=max(t[pos].ls,t[pos<<1].ms+t[pos<<1|1].ls);
t[pos].rs=t[pos<<1|1].rs;
if(t[pos<<1|1].ms==t[pos<<1|1].len) t[pos].rs=max(t[pos].rs,t[pos<<1|1].ms+t[pos<<1].rs);
t[pos].ms=max(t[pos<<1].ms,max(t[pos<<1|1].ms,t[pos<<1].rs+t[pos<<1|1].ls));
}
void build(int pos,int L,int R){
t[pos].L=L;
t[pos].R=R;
t[pos].len=R-L+1;
t[pos].tag=-1;
if(L==R){
t[pos].ls=1;
t[pos].ms=1;
t[pos].rs=1;
return;
}
int mid=(L+R)>>1;
build(pos<<1,L,mid);
build(pos<<1|1,mid+1,R);
pushup(pos);
}
void pushdown(int pos){
if(t[pos].tag!=-1){
t[pos<<1].ls=t[pos<<1].rs=t[pos<<1].ms=t[pos<<1].len*t[pos].tag;
t[pos<<1|1].ls=t[pos<<1|1].rs=t[pos<<1|1].ms=t[pos<<1|1].len*t[pos].tag;
t[pos<<1].tag=t[pos<<1|1].tag=t[pos].tag;
t[pos].tag=-1;
}
}
void update(int pos,int ll,int rr){
int L=t[pos].L,R=t[pos].R;
if(ll>R||L>rr) return;
if(ll<=L&&rr>=R){
t[pos].ls=t[pos].len;
t[pos].rs=t[pos].len;
t[pos].ms=t[pos].len;
t[pos].tag=1;
return;
}
pushdown(pos);
int mid=(L+R)>>1;
update(pos<<1,ll,rr);
update(pos<<1|1,ll,rr);
pushup(pos);
}
void cl(int pos,int ll,int rr){
int L=t[pos].L,R=t[pos].R;
if(ll>R||L>rr) return;
if(ll<=L&&rr>=R){
t[pos].ls=t[pos].rs=0;
t[pos].ms=0;
t[pos].tag=0;
return;
}
pushdown(pos);
cl(pos<<1,ll,rr);
cl(pos<<1|1,ll,rr);
pushup(pos);
}
void query(int pos,int k){
pushdown(pos);
if(t[pos].len<k||t[pos].ms<k) return;
if(t[pos].ls>=k){
loc=t[pos].L;
cl(1,t[pos].L,t[pos].L+k-1);
return;
}
else if(t[pos<<1].ms>=k) query(pos<<1,k);
else if(t[pos<<1].rs+t[pos<<1|1].ls>=k){
loc=t[pos<<1].R-t[pos<<1].rs+1;
cl(1,loc,loc+k-1);
return;
}
else query(pos<<1|1,k);
}
int main(){
scanf("%d%d",&n,&m);
build(1,1,n);
int op,x,y;
for(int i=1;i<=m;++i){
scanf("%d",&op);
if(op==1){
scanf("%d",&x);
if(t[1].ms<x){
printf("0\n");
}
else {
query(1,x);
printf("%d\n",loc);
}
}
else{
scanf("%d%d",&x,&y);
update(1,x,x+y-1);
}
}
return 0;
}