【POJ 3667 】 线段树之区间操作

题目连接：http://poj.org/problem?id=3667

 

题目大意：让你对一个区间进行操作。输入Q C 或者 Q C D。

Q ==1  输入 C： 表示让你求1-n中是否有连续的C个空hotel，如果有多个连续的C个空hotel，则取最左边的，并输出最左边的第一间hotel标号。让人住进去，那么这些空房就不能住人了。如果不存在连续的C个hotel，则输出 0。

Q==2   输入 C D： 表示 从标号C到C+D-1的hotel的房间全部要退房，那么这么hotel都变成空。

 

解题思路：

用到线段树的区间操作，具体解析见代码。

  1 #include <cstdio>

  2 #include <cmath>

  3 #include <iostream>

  4 #include <algorithm>

  5 using namespace std;

  6 

  7 #define lz 2*u,l,mid

  8 #define rz 2*u+1,mid+1,r

  9 const int maxn=50005;

 10 int  flag[4*maxn];   ///标记

 11 

 12 struct node

 13 {

 14     int lm; ///从左边第一个点开始最长的连续空hotel

 15     int rm; ///以右边最后一个结束的最长的连续空hotel

 16     int sm; ///整段区间最大的连续空hotel 

 17 } tree[4*maxn];

 18 

 19 void push_up(int u, int l, int r)   ///向上更新

 20 {

 21     tree[u].lm=tree[2*u].lm;          

 22     tree[u].rm=tree[2*u+1].rm;

 23     tree[u].sm=max(tree[2*u].sm,tree[2*u+1].sm);

 24     int mid=(l+r)>>1;

 25     if(tree[2*u].lm==mid-l+1) tree[u].lm+=tree[2*u+1].lm;  ///!!这里注意，当左孩子左边连续的达到整个区间时，要加上右孩子的左边区间

 26     if(tree[2*u+1].rm==r-mid) tree[u].rm+=tree[2*u].rm;   ///!!考虑右区间，同上

 27     int t=tree[2*u].rm+tree[2*u+1].lm;

 28     if(t>tree[u].sm) tree[u].sm=t;

 29 }

 30 

 31 void push_down(int u, int l, int r)  ///向下更新

 32 {

 33     if(flag[u]==-1) return ;

 34     if(flag[u])

 35     {

 36         flag[2*u]=flag[2*u+1]=flag[u];

 37         tree[2*u].lm=tree[2*u].rm=tree[2*u].sm=0;

 38         tree[2*u+1].lm=tree[2*u+1].rm=tree[2*u+1].sm=0;

 39         flag[u]=-1;

 40     }

 41     else

 42     {

 43         flag[2*u]=flag[2*u+1]=flag[u];

 44         int mid=(l+r)>>1;

 45         tree[2*u].lm=tree[2*u].rm=tree[2*u].sm=mid-l+1;

 46         tree[2*u+1].lm=tree[2*u+1].rm=tree[2*u+1].sm=r-mid;

 47         flag[u]=-1;

 48     }

 49 }

 50 

 51 void build(int u, int l, int r)  ///建树

 52 {

 53     flag[u]=-1;

 54     if(l==r)

 55     {

 56         tree[u].lm=tree[u].rm=tree[u].sm=1;

 57         return ;

 58     }

 59     int mid=(l+r)>>1;

 60     build(lz);

 61     build(rz);

 62     push_up(u,l,r);

 63 }

 64 

 65 void Update(int u, int l, int r, int tl, int tr, int c)   ///更新操作

 66 {

 67     if(tl<=l&&r<=tr)

 68     {

 69         tree[u].sm=tree[u].lm=tree[u].rm=(c==1?0:r-l+1);

 70         flag[u]= c;  

 71         return ;

 72     }

 73     push_down(u,l,r);   ///再次遇见此段区间时，延迟标记同步向下更新

 74     int mid=(l+r)>>1;

 75     if(tr<=mid) Update(lz,tl,tr,c);

 76     else if(tl>mid) Update(rz,tl,tr,c);

 77     else

 78     {

 79         Update(lz,tl,mid,c);    ///注意区间分隔开，tl，tr跨越两个左右区间

 80         Update(rz,mid+1,tr,c);

 81     }

 82     push_up(u,l,r);     ///递归的时候同步向上更新

 83 }

 84 

 85 int Query(int u, int l, int r, int num)   ///询问操作

 86 {

 87     if(l==r)

 88         return l;

 89     push_down(u,l,r);     ///延迟标记向下传递

 90     int mid=(l+r)>>1;

 91     if(tree[2*u].sm>=num) return Query(lz,num);

 92     else if(tree[2*u].rm+tree[2*u+1].lm>=num&&tree[2*u].rm>=1) return mid-tree[2*u].rm+1;   ///满足条件时，返回左边rm连续的hotel第一个房间标号

 93     else

 94         return Query(rz,num);

 95 }

 96 

 97 int main()

 98 {

 99     int n, m;

100     while(~scanf("%d%d",&n,&m))

101     {

102         build(1,1,n);

103         while(m--)

104         {

105             int p, u, v;

106             scanf("%d",&p);

107             if(p==1)

108             {

109                 scanf("%d",&u);

110                 if(tree[1].sm<u)  ///特判一下是否有这么多个连续的空hotel，没有则直接输出，不用操作

111                 {

112                     puts("0"); continue;

113                 }

114                 int p=Query(1,1,n,u);

115                 printf("%d\n",p);

116                 Update(1,1,n,p,p+u-1,1);

117             }

118             else

119             {

120                 scanf("%d%d",&u,&v);

121                 Update(1,1,n,u,u+v-1,0);

122             }

123         }

124     }

125     return 0;

126 }