题目大意:
有一排标号1-N的房间。
操作一:询问是不是有连续长度为a的空房间,有的话住进最左边(占用a个房间)
操作二:将[a,a+b-1]的房间清空(腾出b个房间)
思路:记录每个区间中“靠左”“靠右”“中间”的空房间
线段树操作:update:区间替换
query:询问满足条件的最左端点
题目链接:
http://vjudge.net/problem/viewProblem.action?id=10354
题目操作:
因为这里找从最左边住起的房间,所以这里不能像常规地写query函数那样写了,终于发现自己只会套模板的下场的悲哀了,智商拙计啊T T
而且你在query函数,因为要尽可能找左边的房间,所以要从左边先递归
1.
int query(int cur,int x,int y,int w)
{
int mid=(x+y)>>1,ls=cur<<1,rs=cur<<1|1;
if(x==y) return x; //这里是表示找到树的底层无叶节点了就结束函数,同时也防止找不到点,其实这里如果找不到所求点的话,会不断进入右子树最后抵达最大的子节点返回
pushdown(cur,x,y);
if(mc[cur<<1]>=w) return query(ls,x,mid,w); //如果左侧能找到满足的点,我们不断朝左侧进行递归
else if(rc[cur<<1]+lc[cur<<1|1]>=w) return mid-rc[cur<<1]+1;//左侧不成立我们在开始找中间并在一起形成的房间的最左侧点
return query(rs,mid+1,y,w); //左侧中间都不成立,那么只能进入右子树进行找点
}
因为这个query函数必然有返回值,所以,我们在主函数中必须先进行判断能否找到适合的一连串的房间,然后再进行ans=query(1,1,n,w)操作以及后面的房间入住的覆盖操作。
这个判断很简单 1为线段树根节点,所以mc[1]其实是最大的连续最长房间,if(mc[1]>=w)这个判断执行就可以了
这里介绍一下puts("0"):
puts()函数用来向标准输出设备(屏幕)写字符串并换行,其调用方式为,puts(s);其中s为字符串字符(字符串数组名或字符串指针)。
所以这道题puts("0");就输出了不存在的情况
2.
当然我也可以换种方式在query的函数里面进行一下小小的修改,使得它在不能找到房间的时候输出-1;
int query(int cur,int x,int y,int w)
{
int mid=(x+y)>>1,ls=cur<<1,rs=cur<<1|1;
if(x==y) return mc[cur]<w?-1:x; //在这修改一下,然后直接在main函数中判断为-1,那么输出0即可
pushdown(cur,x,y);
if(mc[cur<<1]>=w) return query(ls,x,mid,w);
else if(rc[cur<<1]+lc[cur<<1|1]>=w) return mid-rc[cur<<1]+1;
return query(rs,mid+1,y,w);
}
总代码如下:
1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 using namespace std; 5 #define N 50010 6 #define L ls,x,mid 7 #define R rs,mid+1,y 8 int rev[N<<2],lc[N<<2],rc[N<<2],mc[N<<2]; 9 void update(int cur,int x,int y) 10 { 11 int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1; 12 lc[cur]=lc[ls],rc[cur]=rc[rs]; 13 mc[cur]=max(mc[ls],mc[rs]); 14 mc[cur]=max(mc[cur],rc[ls]+lc[rs]); 15 if(lc[ls]==mid-x+1) lc[cur]=lc[ls]+lc[rs]; 16 if(rc[rs]==y-mid) rc[cur]=rc[rs]+rc[ls]; 17 } 18 void build(int cur,int x,int y) 19 { 20 int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1; 21 rev[cur]=-1; 22 if(x==y){ 23 lc[cur]=rc[cur]=mc[cur]=1; 24 return; 25 } 26 build(ls,x,mid); 27 build(rs,mid+1,y); 28 update(cur,x,y); 29 } 30 void pushdown(int cur,int x,int y) 31 { 32 int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1; 33 if(rev[cur]!=-1){ 34 if(rev[cur]==1){ 35 rev[ls]=rev[rs]=1; 36 lc[ls]=rc[ls]=mc[ls]=lc[rs]=rc[rs]=mc[rs]=0; 37 rev[cur]=-1; 38 } 39 else if(rev[cur]==0){ 40 rev[ls]=rev[rs]=0; 41 lc[ls]=rc[ls]=mc[ls]=mid-x+1; 42 lc[rs]=rc[rs]=mc[rs]=y-mid; 43 rev[cur]=-1; 44 } 45 } 46 } 47 void change(int cur,int x,int y,int s,int t,int v) 48 { 49 int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1; 50 if(x>=s&&y<=t){ 51 rev[cur]=v; 52 lc[cur]=rc[cur]=mc[cur]=v?0:y-x+1; 53 return; 54 } 55 pushdown(cur,x,y); 56 if(mid>=s) change(ls,x,mid,s,t,v); 57 if(mid+1<=t) change(rs,mid+1,y,s,t,v); 58 update(cur,x,y); 59 } 60 int query(int cur,int x,int y,int w) 61 { 62 int mid=(x+y)>>1,ls=cur<<1,rs=cur<<1|1; 63 if(x==y) return x; 64 pushdown(cur,x,y); 65 if(mc[cur<<1]>=w) return query(ls,x,mid,w); 66 else if(rc[cur<<1]+lc[cur<<1|1]>=w) return mid-rc[cur<<1]+1; 67 return query(rs,mid+1,y,w); 68 } 69 int main() 70 { 71 int n,m,op,a,b; 72 while(scanf("%d%d",&n,&m)!=EOF){ 73 build(1,1,n); 74 for(int i=0;i<m;i++){ 75 scanf("%d",&op); 76 if(op==1){ 77 scanf("%d",&a); 78 if(mc[1]<a) puts("0"); 79 else{ 80 int ans=query(1,1,n,a); 81 printf("%d\n",ans); 82 change(1,1,n,ans,ans+a-1,1); 83 } 84 } 85 else{ 86 scanf("%d%d",&a,&b); 87 change(1,1,n,a,a+b-1,0); 88 } 89 } 90 } 91 return 0; 92 }
POJ 3667 线段树的区间合并简单问题