试题名称:公共钥匙盒
问题描述
有一个学校的老师共用N个教室,按照规定,所有的钥匙都必须放在公共钥匙盒里,老师不能带钥匙回家。每次老师上课前,都从公共钥匙盒里找到自己上课的教室的钥匙去开门,上完课后,再将钥匙放回到钥匙盒中。
钥匙盒一共有N个挂钩,从左到右排成一排,用来挂N个教室的钥匙。一串钥匙没有固定的悬挂位置,但钥匙上有标识,所以老师们不会弄混钥匙。
每次取钥匙的时候,老师们都会找到自己所需要的钥匙将其取走,而不会移动其他钥匙。每次还钥匙的时候,还钥匙的老师会找到最左边的空的挂钩,将钥匙挂在这个挂钩上。如果有多位老师还钥匙,则他们按钥匙编号从小到大的顺序还。如果同一时刻既有老师还钥匙又有老师取钥匙,则老师们会先将钥匙全还回去再取出。
今天开始的时候钥匙是按编号从小到大的顺序放在钥匙盒里的。有K位老师要上课,给出每位老师所需要的钥匙、开始上课的时间和上课的时长,假设下课时间就是还钥匙时间,请问最终钥匙盒里面钥匙的顺序是怎样的?
输入格式
输入的第一行包含两个整数N, K。
接下来K行,每行三个整数w, s, c,分别表示一位老师要使用的钥匙编号、开始上课的时间和上课的时长。可能有多位老师使用同一把钥匙,但是老师使用钥匙的时间不会重叠。
保证输入数据满足输入格式,你不用检查数据合法性。
输出格式
输出一行,包含N个整数,相邻整数间用一个空格分隔,依次表示每个挂钩上挂的钥匙编号。
样例输入
5 2
4 3 3
2 2 7
样例输出
1 4 3 2 5
样例说明
第一位老师从时刻3开始使用4号教室的钥匙,使用3单位时间,所以在时刻6还钥匙。第二位老师从时刻2开始使用钥匙,使用7单位时间,所以在时刻9还钥匙。
每个关键时刻后的钥匙状态如下(X表示空):
时刻2后为1X345;
时刻3后为1X3X5;
时刻6后为143X5;
时刻9后为14325。
样例输入
5 7
1 1 14
3 3 12
1 15 12
2 7 20
3 18 12
4 21 19
5 30 9
样例输出
1 2 3 5 4
评测用例规模与约定
对于30%的评测用例,1 ≤ N, K ≤ 10, 1 ≤ w ≤ N, 1 ≤ s, c ≤ 30;
对于60%的评测用例,1 ≤ N, K ≤ 50,1 ≤ w ≤ N,1 ≤ s ≤ 300,1 ≤ c ≤ 50;
对于所有评测用例,1 ≤ N, K ≤ 1000,1 ≤ w ≤ N,1 ≤ s ≤ 10000,1 ≤ c ≤ 100。
分析:
用一个数组box表示钥匙盒,数组的下标表示挂钩的位置,数组的内容表示挂钩上钥匙的编号,创建一个结构体表示事件,包含钥匙编号,以及时间结点,和flag标记表示取还是拿。
把事件压入优先队列。首先时间结点的值越小优先级越大,如果时间相同则flag的值越小优先级越大(先还再拿),若flag与时间都相同则key的值越小优先级越大(按从小到大顺序先还再拿)
优先队列都存储好之后取队列顶部元素,每出列一个就对Box进行一次操作,而后弹出,取下一个队顶元素。
最后输出即可。
代码如下:
1 #include<iostream> 2 #include<queue> 3 using namespace std; 4 struct Node{ 5 int key; 6 int time; 7 int flag; 8 bool operator <(Node a) const 9 { 10 if(time!=a.time) return time>a.time;//小的优先级高 11 else if(flag!=a.flag) return flag>a.flag; 12 else return key>a.key; 13 } 14 }; 15 priority_queue<Node> q; 16 const int maxN=1000; 17 int main(){ 18 int N,K,w,s,c; 19 cin>>N>>K; 20 int box[maxN+1]; 21 for(int i=1;i<N+1;i++){ 22 box[i]=i; 23 } 24 Node event1[K]; 25 Node event2[K]; 26 Node temp; 27 for(int i=0;i<K;i++){ 28 cin>>w>>s>>c; 29 event1[i].key=w; 30 event1[i].time=s; 31 event1[i].flag=1; 32 event2[i].key=w; 33 event2[i].time=s+c; 34 event2[i].flag=-1; 35 q.push(event1[i]); 36 q.push(event2[i]); 37 } 38 while(!q.empty()){ 39 temp=q.top(); 40 q.pop(); 41 if(temp.flag==1){ 42 int j=1; 43 while(box[j]!=temp.key) j++; 44 box[j]=0; 45 } 46 else { 47 int j=1; 48 while(box[j]!=0) j++; 49 box[j]=temp.key; 50 } 51 } 52 cout<<box[1]; 53 for(int i=2;i<N+1;i++){ 54 cout<<" "<<box[i]; 55 } 56 cout<<endl; 57 return 0; 58 }