首页 > 代码库 > POJ 3667 线段树的区间合并简单问题

POJ 3667 线段树的区间合并简单问题

题目大意:
有一排标号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 }