NYOJ250 ROOM ASSIGNATION || POJ3667 Hotel
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 }