参考的博客
题目描述
有一个学校的老师共用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。
C++代码
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
//存储老师上课的输入信息
typedef struct{
//钥匙,上课时刻,上课时间
int w,s,c;
}info;
typedef struct time{
int w,c;//c为下课时刻
bool operator < (const time &rhs) const{
if(c == rhs.c)//如果下课时间相同就按照钥匙序号从小到大排序
return w>rhs.w;
return c>rhs.c;
}
}time;
bool cmp(info i1,info i2){
return i1.s<i2.s;
}
int main(int argc, char** argv) {
int N,K;
int res[1002];//存放钥匙
info in[1002];//存储老师上课信息
priority_queue<time> Q;
cin>>N>>K;
for(int i=1;i<=N;i++){
res[i]=i;
}
for(int i=0;i<K;i++){
cin>>in[i].w>>in[i].s>>in[i].c;
}
sort(in,in+K,cmp);
for(int i=0;i<K;i++){
time t;
t.w = in[i].w;
t.c = in[i].c+in[i].s;//下课时刻
while(!Q.empty()&&Q.top().c<=in[i].s){
/*如果在优先队列中队首的下课时刻小于等于本次循环的上课时刻
说明该还钥匙了*/
//循环找到第一个-1,即最左边,放入钥匙
for(int j=1;j<=N;j++){
if(res[j]==-1){
res[j]=Q.top().w;
break;
}
}
Q.pop();
}
Q.push(t);
for(int j=1;j<=N;j++){
//数组值等于加入队列元素的w,修改值表示取走钥匙
if(res[j]==in[i].w){
res[j]=-1;
break;
}
}
}
for(int j=1;j<=N;j++)
{
if(res[j]==-1){
res[j]=Q.top().w;
Q.pop();
}
cout<<res[j]<<" ";
}
return 0;
}