CCF 201709-2 公共钥匙盒(100) C/C++

题目

试题编号： 201709-2
试题名称： 公共钥匙盒
时间限制： 1.0s
内存限制： 256.0MB
问题描述：
问题描述
　　有一个学校的老师共用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。

思路

定义一个结构体，包括钥匙ID，开始时间，归还时间。在k行输入时统计需要操作的时间节点（运用hash的思想），这有利于降低时间复杂度，博主第一次提交代码时60分，根据评测用例规模与约定，显然当N,K,w,s,c规模过大时，运行超时。代码使用了algorithm头文件里的sort函数对cla数组里的钥匙ID进行排序，便于多人归还时的顺序问题。这里提供另外一种思路，在统计时间节点时，既可以将需要操作的钥匙ID链在该时间节点，就不用在之后循环查找了。

代码

#include <iostream>
#include <algorithm>
using namespace std;
int tmp[10101],time[10101];
struct teacher{
	int id;     //钥匙编号 
	int start;  //开始时间 
	int over;   //归还时间 
};
bool cmp(teacher a,teacher b){      //sort函数的cmp，将钥匙编号从小到大排序，便于实现当多位老师同时归还时的先后顺序 
	return a.id <b.id;
}
int main(){
	int n,k;
	cin>>n>>k;
	int key[2][n+1];   //钥匙盒 1为有
	for(int i=1;i<=n;i++){
		key[0][i]=i;
		key[1][i]=1;
	}
	struct teacher cla[k];
	fill(tmp,tmp+10101,0);     //tmp和time的作用是降低时间复杂度 
	fill(time,time+10101,0);
	for(int i=0;i<k;i++){
		cin>>cla[i].id>>cla[i].start>>cla[i].over;
		cla[i].over+=cla[i].start;
		tmp[cla[i].start]++;    //运用hash的思想，统计哪些时间点有归还或使用的操作 
		tmp[cla[i].over]++;
	} 
	int t=0;
	for(int i=0;i<10101;i++){    
		if(tmp[i])
		time[t++]=i;           //从time[0]开始记录需要操作的时间节点 
	}
	sort(cla,cla+k,cmp);       //钥匙编号从小到大排序 
	for(int i=0;i<t;i++){
		for(int j=0;j<k;j++){         //先进行归还操作，这里找到的第一个必是钥匙ID最小的 
			if(cla[j].over==time[i]){
				for(int r=1;r<=n;r++){         //从1到n查找空钥匙盒 
					if(key[1][r]==0){
						key[1][r]=1;
						key[0][r]=cla[j].id;
						break;               //归还后结束循环 
					}
				}
			}
		}
		for(int j=0;j<k;j++){               //取钥匙操作 
			if(cla[j].start==time[i]){
				for(int r=1;r<=n;r++){
					if(cla[j].id==key[0][r])
					key[1][r]=0;
				}
			}
		}
	}
	for(int i=1;i<=n;i++)
	cout<<key[0][i]<<" ";
	return 0;
}