CCF 201709-2 公共钥匙盒

题目

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

 

思路

首先如果以老师为单位考虑问题的话会变得很复杂，借钥匙时间、还钥匙时间、钥匙摆放顺序都比较难放在一起考虑，所以可以把借和还拆分成两个动作，根据题中要求，借动作的时间就是我们输入的借钥匙时间，还钥匙的时间是借钥匙的时间加上上课时长，再通过引入一个标志变量将借动作和还动作加以区分，再对存储了动作的数组进行排序，如果时间不一致则按照先后顺序就可以了，如果时间相同，则需要先执行还动作后执行借动作，如果动作也都是一致的，则需要根据动作执行的对象——钥匙的编号，从小到大执行，基于这种想法，建立了一个函数，再通过调用sort函数实现对数组的排序

bool cmp(struct key k1,struct key k2)
{
	if(k1.time!=k2.time)//时间不同则按先后顺序 
		return k1.time<k2.time;
	else if(k1.sta!=k2.sta)//时间顺序相同先还后借 
		return k1.sta>k2.sta;
	else//动作相同则按照序号顺序 
		return k1.id<k2.id;
 } 

代码

#include <iostream>
#include <algorithm>
using namespace std;

struct key
{
	int id;//钥匙编号 
	int sta;//状态   已还/在借  还的优先级大于借   借1 还2 
	int time;//借钥匙的时间或者还钥匙的时间 
}; 

bool cmp(struct key k1,struct key k2)
{
	if(k1.time!=k2.time)//时间不同则按先后顺序 
		return k1.time<k2.time;
	else if(k1.sta!=k2.sta)//时间顺序相同先还后借 
		return k1.sta>k2.sta;
	else//动作相同则按照序号顺序 
		return k1.id<k2.id;
 } 

int main()
{
	int n,k;
	cin>>n>>k;
	int order[n];
	for(int i=1;i<=n;i++)//钥匙盒中编号顺序 
		order[i-1]=i;
	struct key que[2*k];
	for(int i=0;i<k;i++)
	{
		int id,start,end;
		cin>>id>>start>>end;
		que[i*2].id=id;
		que[i*2].time=start;
		que[i*2].sta=1;
		que[i*2+1].id=id;
		que[i*2+1].sta=2;
		que[i*2+1].time=start+end;
	}
	sort(que,que+2*k,cmp);
	/*for(int i=0;i<2*k;i++)
	{
		cout<<que[i].id<<" "<<que[i].sta<<' '<<que[i].time<<endl;
	}*/
	for(int i=0;i<2*k;i++)
	{
		if(que[i].sta==1)
		{
			for(int j=0;j<n;j++)
			{
				if(order[j]==que[i].id)
				{
					order[j]=0;
					break;
				}
			}
		}
		else if(que[i].sta==2)
		{
			for(int j=0;j<n;j++)
			{
				if(order[j]==0)
				{
					order[j]=que[i].id;
					break;
				}
			}
		}
	    /*for(int ii=0;ii<n;ii++)
		{
			cout<<order[ii];
		}
		cout<<endl;*/
	}
	for(int i=0;i<n;i++)
	{
		if(i!=n-1)
			cout<<order[i]<<' ';
		else
			cout<<order[i];
	}
	return 0;
}

运行结果