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 }