1 /*
2 Name:
3 Copyright:
4 Author:
5 Date: 28/04/13 21:55
6 Description:
7 刚开始做这道题的时候，第一感觉用线段树应该很容易实现，结果发现我没做过此类的线段树题目
8 然后上网看了别人的代码，才了有点思路，现总结如下：
9 这道题题意就是旅馆订房问题，由于牵扯到某一段区间的查找与运算，所以选择了线段树求解，
10 用free表示此段区间房间是否可用，lmax代表这段区间从左边起最长空闲房间数，rmax表示这段区间
11 从右边起最长空闲房间数，tmax表示整个区间上的最长连续空闲房间数，查找连续空房间数时，过程如下：
12 1、如果1~n区间的tmax值都比待查长度len小，就无解，否则有解执行2
13 2、假如区间lmax是否大于等于len，直接返回l ，否则执行3
14 3、对于每一个区间，如果它的左儿子的tmax值大于等于len，到左儿子里去找。否则执行4
15 4、如果左儿子的rmax加上有儿子的lmax大于等于len，直接返回左儿子的右端点减去左儿子的rmax值。
16 5、否则到右儿子里去找。
17 PS：对于上述过程，如果满足前面的条件，则不再考虑后面的情况。
18 更新的时候既进行下压操作，在对某段区间进行处理后（清空房间或占用房间）后，进行更新父节点，
19 细节看注释！！
20 */
21 #include<iostream>
22 #include<cstdio>
23 const int N=50010;
24 using namespace std;
25 struct Node{
26 int l,r;
27 int free;
28 int lmax,rmax,tmax;
29 }tree[3*N];
30 int build(int l,int r,int i){
31 tree[i].l=l;
32 tree[i].r=r;
33 tree[i].lmax=tree[i].rmax=tree[i].tmax=r-l+1; //初始化为区间大小
34 tree[i].free=0;
35 if(l<r){
36 int mid=(l+r)>>1;
37 build(l,mid,i<<1);
38 build(mid+1,r,i<<1|1);
39 }
40 }
41 int query(int i,int len){
42 if(tree[i].tmax<len) return 0; //情况1
43 if(tree[i].lmax>=len) return tree[i].l; //情况2
44 if(tree[i<<1].tmax>=len) return query(i<<1,len); //情况3
45 else if(tree[i<<1].rmax+tree[i<<1|1].lmax >= len) //情况4
46 return tree[i<<1].r-tree[i<<1].rmax+1;
47 else return query(i<<1|1,len); //情况5
48 }
49 int update(int l,int r,int i,int state){
50 if(tree[i].l==l&&tree[i].r==r){ //找到此段区间
51 tree[i].free=state;
52 if(state){ //若为1则占用房间 ，否则腾出房间
53 tree[i].lmax=tree[i].rmax=tree[i].tmax=0;
54 }else{
55 tree[i].lmax=tree[i].rmax=tree[i].tmax=tree[i].r-tree[i].l+1;
56 }
57 }else if(tree[i].r > tree[i].l){
58 if(tree[i].free==1){ //树节点的相应信息向下压，即节点信息细致化
59 tree[i<<1].free=tree[i<<1|1].free=1;
60 tree[i<<1].lmax=tree[i<<1].rmax=tree[i<<1].tmax=0;
61 tree[i<<1|1].lmax=tree[i<<1|1].rmax=tree[i<<1|1].tmax=0;
62 }
63 if(tree[i].free==0){
64 tree[i<<1].free=tree[i<<1|1].free=0;
65 tree[i<<1].lmax=tree[i<<1].rmax=tree[i<<1].tmax=tree[i<<1].r-tree[i<<1].l+1;
66 tree[i<<1|1].lmax=tree[i<<1|1].rmax=tree[i<<1|1].tmax=tree[i<<1|1].r-tree[i<<1|1].l+1;
67 }
68 int mid=(tree[i].r+tree[i].l)>>1;
69 if(mid>=r) update(l,r,i<<1,state);
70 else if(mid<l) update(l,r,i<<1|1,state);
71 else{
72 update(l,mid,i<<1,state);
73 update(mid+1,r,i<<1|1,state);
74 }
75 tree[i].lmax=tree[i<<1].lmax; //树节点相应信息向上推送，即更新父节点信息
76 tree[i].rmax=tree[i<<1|1].rmax;
77 if(tree[i<<1].free==0) tree[i].lmax+=tree[i<<1|1].lmax;
78 if(tree[i<<1|1].free==0) tree[i].rmax+=tree[i<<1].rmax;
79 tree[i].tmax=max(tree[i<<1].tmax,tree[i<<1|1].tmax); //选取左孩子tmax和右孩子tmax及左右之间结合的最大值
80 tree[i].tmax=max(tree[i].tmax,tree[i<<1].rmax+tree[i<<1|1].lmax);
81 if(tree[i<<1].free==tree[i<<1|1].free) //更新父节点的使用情况
82 tree[i].free=tree[i<<1].free;
83 else tree[i].free=-1;
84 }
85 }
86 int main()
87 {
88 int N,M,s,num,start,flag;
89 while(~scanf("%d%d",&N,&M)){
90 build(1,N,1);
91 while(M--){
92 scanf("%d",&flag);
93 if(flag==1){
94 scanf("%d",&s);
95 start=query(1,s); //start为0说明未找到符合房间
96 printf("%d\n",start);
97 if(start) update(start,start+s-1,1,1); //假如 start不为零，那么就需要占用
98 }else{
99 scanf("%d%d",&s,&num);
100 update(s,s+num-1,1,0);
101 }
102 }
103 }
104 return 0;
105 }
浙公网安备 33010602011771号