线段树 洛谷P2894 [USACO08FEB]酒店Hotel
https://www.luogu.org/problem/show?pid=2894#sub
这道题第一天我觉得贼难,感觉很复杂;
第二天看便有了思路,但觉得写起来不简单;
第三天发现,只不过是线段树里面多维护了几个东西罢了;
以后的练习,估计线段树不再仅仅是维护一下数值就好了;
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#define Ll long long
using namespace std;
struct tree{
int xx,yy,l,r,m,p;//xx,yy是这个区间的范围,l,r代表这段区间的左右最长连续空位,m就是整段区间的连续空位,p是lazy标记
}T[500000*8+1];
int n,m,x,y,z,ans;
void clean(int num){//这个显然就是lazy的下传
if(T[num].xx==T[num].yy)return;
if(!T[num].p)return;
int k=num+num;
if(T[num].p==1){
T[k].l=T[k].m=T[k].r=T[k+1].l=T[k+1].m=T[k+1].r=0;
T[k].p=T[k+1].p=1;
}else{
T[k].l=T[k].m=T[k].r=T[k].yy-T[k].xx+1;
T[k+1].l=T[k+1].m=T[k+1].r=T[k+1].yy-T[k+1].xx+1;
T[k].p=T[k+1].p=2;
}
T[num].p=0;
}
void maketree(int x,int y,int num){
T[num].xx=x;T[num].yy=y;
T[num].l=T[num].r=T[num].m=y-x+1;//一开始全空房
if(x==y)return;
int mid=(x+y)>>1;num=num<<1;
maketree(x,mid ,num );
maketree(mid+1,y,num+1);
}
int find(int x,int num){//找一个长度位x的区间,现搜索到第num个区间
clean(num);clean(num*2);clean(num*2+1);
if(T[num].l>=x)return T[num].xx;//如果这个区间的左连续区间足够,那么直接住在最左边吧
if(T[num*2].m>=x)return find(x,num*2);//如果左半段区间的最大值满足x,说明可以住在左子儿子的区间
if(T[num*2].r+T[num*2+1].l>=x)return T[num*2].yy-T[num*2].r+1;//这时判断中间的一块区域
return find(x,num*2+1); //因为一定可以找到区间,但前面都找不到,那只可能在右半段的区间里了
}
void gaoji(int num){//ohly~ohlyohlyohly~~ wearethegay~~~wearethegay~~~~~~~~随便取的函数命
int k=num<<1;
clean(k);clean(k+1);
T[num].l=T[k ].l;//显然大区间的l是左小区间的l,但是如果左小区间全空,显然大区间的l要加上右小区间的l
if(T[k ].l==T[k ].yy-T[k ].xx+1)T[num].l+=T[k+1].l;
T[num].r=T[k+1].r;//同理
if(T[k+1].r==T[k+1].yy-T[k+1].xx+1)T[num].r+=T[k ].r;
T[num].m=max(T[k].m,T[k+1].m);//这个很显然啊
T[num].m=max(T[num].m,T[k].r+T[k+1].l);
}
void dec(int x,int y,int num){//这些和基本线段树差不多
clean(num);
if(x<=T[num].xx&&T[num].yy<=y){T[num].l=T[num].r=T[num].m=0;T[num].p=1;return;}
num=num<<1;
if(x<=T[num ].yy)dec(x,y,num );
if(y>=T[num+1].xx)dec(x,y,num+1);
gaoji(num/2);
}
void pluss(int x,int y,int num){
clean(num);
if(x<=T[num].xx&&T[num].yy<=y){T[num].l=T[num].r=T[num].m=T[num].yy-T[num].xx+1;T[num].p=2;return;}
num=num<<1;
if(x<=T[num ].yy)pluss(x,y,num );
if(y>=T[num+1].xx)pluss(x,y,num+1);
gaoji(num/2);
}
int main()
{
scanf("%d%d",&n,&m);
maketree(1,n,1);
while(m--){
scanf("%d",&x);
if(x==1){
scanf("%d",&x);
if(T[1].m<x){printf("0\n");continue;}//T[1].m是整个区间中的最大连续空位,这个判断显然成立
ans=find(x,1);//通过了上面一个判断,一定可以找出答案
printf("%d\n",ans);
dec(ans,ans+x-1,1);//把新住进来的房间重置
}else{
scanf("%d%d",&x,&y);
pluss(x,x+y-1,1);//把新走出去的房间重置
}
}
}