CCF 201709-2 公共钥匙盒 从构思到写完所花时间大概45分钟。

CCF 201709-2 公共钥匙盒  写完所花时间大概45分钟。

 

 

问题描述

　　有一个学校的老师共用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。

 

代码如下，注释已经打好，欢迎指出不足：

整体思路主要就是空间换时间+适当剪枝。

//空间换时间 
#include<iostream>
#include<vector>
#include<algorithm>

using namespace std; 
vector<int> time[10200][2];//时间队列. 第一维度代表 取(0)/还(1) 钥匙 
int main()
{
	int n;//n把钥匙
	int k;//k个老师

	cin>>n>>k;
	int key[1001];//钥匙队列；
	for(int i=1;i<=n;i++)
	{
		key[i]=i;;
	}
	int w, s, c;//钥匙编号，开始时间，上课时长。
	//剪枝 
	int mint=20000;//最早的时间 
	int maxt=0; //最迟的时间 
	int over;//结束时间 
	for(int i=0;i<k;i++)
	{
		cin>>w>>s>>c;
		over=s+c;
		if(s<mint)
		{
			mint=s;
		}
		if(maxt<over)
		{
			maxt=over;
		}
		time[s][0].push_back(w);//取钥匙 
		time[over][1].push_back(w);//还钥匙
		//如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还。
		sort(time[over][1].begin(),time[over][1].end());//还钥匙按顺序 
	 } 
	 //如果同一时刻既有老师还钥匙又有老师取钥匙，则老师们会先将钥匙全还回去再取出。
	 
	 
	 vector<int>::iterator it;
	 //从mint到maxt遍历 
	 for(int i=mint;i<=maxt;i++)
	 {
	 	//先还钥匙
		 for(it=time[i][1].begin();it!=time[i][1].end();it++ )
		 {
		 	for(int j=1;j<=n;j++)
			{
				if(key[j]==0)
				{
					key[j]=*it;
					//time[i][1].erase(it);//erase的同时it++了 
					break;
				}
			}
		 }
		 //然后借钥匙
		  for(it=time[i][0].begin();it!=time[i][0].end();it++ )
		 {
		 	key[find(key+1,key+n+1,*it)-key]=0;//找到这把钥匙 
		 }
	 }
	 for(int j=1;j<=n;j++)
		{
			cout<<key[j]<<" ";
		}
	  return 0;
	 
}