线段树专辑——pku 3667 Hotel

http://poj.org/problem?id=3667

哈哈，经典中的经典题啊。利用线段树求最大连续空闲区间，并返回空闲区间的起点坐标。

View Code

  1 #include<iostream>
  2 #include<string>
  3 #include<algorithm>
  4 using namespace std;
  5 
  6 struct node
  7 {
  8     int l;
  9     int r;
 10     int l_val; //从左节点数起的最大连续空闲区间
 11     int r_val; //从右节点起的最大连续空闲区间
 12     int max_val; //整个区间的最大连续空闲区间
 13     int cover;  //当前区间的占用情况，1表示占用，0表示不占用，-1表示既有占用又有不占用
 14 }; 
 15 
 16 node tree[250000];
 17 int n,m;
 18 
 19 int max(int a,int b)
 20 {
 21     return a>b?a:b;
 22 }
 23 
 24 void build(int i,int l,int r)
 25 {
 26     tree[i].l=l;
 27     tree[i].r=r;
 28     tree[i].cover=0;
 29     tree[i].l_val=tree[i].r_val=tree[i].max_val=r-l+1; //开始所有区间都未使用
 30     if(l==r)
 31         return;
 32     int mid=(l+r)/2;
 33     build(2*i,l,mid);
 34     build(2*i+1,mid+1,r);
 35 }
 36 
 37 void fun(int i) //跟新区间
 38 {
 39     if(tree[i].cover==0)
 40     {
 41         tree[i].l_val=tree[i].r_val=tree[i].max_val=tree[i].r-tree[i].l+1;    
 42     }
 43     else if(tree[i].cover==1)
 44     {
 45         tree[i].l_val=tree[i].r_val=tree[i].max_val=0;
 46     }
 47 }
 48 
 49 void updata(int i,int l,int r,int w)
 50 {
 51     if(tree[i].l>r || tree[i].r<l)
 52         return;
 53     if(tree[i].l>=l && tree[i].r<=r)
 54     {
 55         tree[i].cover=w; //跟新该段情况
 56         fun(i);  //并根据cover跟新其他域
 57         return;
 58     }
 59     if(tree[i].cover!=-1)
 60     {
 61         tree[2*i].cover=tree[2*i+1].cover=tree[i].cover;
 62         fun(2*i);  fun(2*i+1);
 63         tree[i].cover=-1;
 64     }
 65     updata(2*i,l,r,w);
 66     updata(2*i+1,l,r,w);
 67     if(tree[2*i].cover==tree[2*i+1].cover) //由下而上跟新所有信息
 68         tree[i].cover=tree[2*i].cover;
 69     else
 70         tree[i].cover=-1;
 71     tree[i].l_val=tree[2*i].l_val+(tree[2*i].cover==0 ? tree[2*i+1].l_val:0);
 72     tree[i].r_val=tree[2*i+1].r_val+(tree[2*i+1].cover==0 ? tree[2*i].r_val:0);
 73     tree[i].max_val=max(max(tree[2*i].max_val,tree[2*i+1].max_val),(tree[2*i].r_val+tree[2*i+1].l_val));
 74 }
 75 
 76 int find(int i,int w)
 77 {
 78     if(tree[i].cover==0 && tree[i].l_val>=w) //从左边起能找到
 79     {
 80         return tree[i].l;
 81     }
 82     if(tree[i].cover==1)
 83     {
 84         return 0;
 85     }
 86     if(tree[i].cover==-1 && tree[i].max_val>=w)
 87     {
 88         if(tree[2*i].max_val>=w) //左边还有足够空间，向左递归
 89             return find(2*i,w);
 90         else if(tree[2*i].r_val+tree[2*i+1].l_val>=w) //如果中间有足够空间，直接计算
 91         {
 92             return tree[2*i].r-tree[2*i].r_val+1;
 93         }
 94         else
 95             return find(2*i+1,w); //向右递归
 96     }
 97     return 0;
 98 }
 99 
100 int main()
101 {
102     freopen("D:\\in.txt","r",stdin);
103     int i,a,b,c;
104     while(scanf("%d%d",&n,&m)!=EOF)
105     {
106         build(1,1,n);
107         for(i=0;i<m;i++)
108         {
109             scanf("%d",&c);
110             if(c==1)
111             {
112                 scanf("%d",&a);
113                 b=find(1,a);
114                 printf("%d\n",b);
115                 if(b)
116                 {
117                     updata(1,b,b+a-1,1);
118                 }
119             }
120             else
121             {
122                 scanf("%d%d",&a,&b);
123                 updata(1,a,a+b-1,0);
124             }
125         }
126     }
127     return 0;
128 }