题目
问题描述 有一个学校的老师共用N个教室,按照规定,所有的钥匙都必须放在公共钥匙盒里,老师不能带钥匙回家。每次老师上课前,都从公共钥匙盒里找到自己上课的教室的钥匙去开门,上完课后,再将钥匙放回到钥匙盒中。 输入格式 输入的第一行包含两个整数N, K。 输出格式 输出一行,包含N个整数,相邻整数间用一个空格分隔,依次表示每个挂钩上挂的钥匙编号。 样例输入 5 2 样例输出 1 4 3 2 5 样例说明 第一位老师从时刻3开始使用4号教室的钥匙,使用3单位时间,所以在时刻6还钥匙。第二位老师从时刻2开始使用钥匙,使用7单位时间,所以在时刻9还钥匙。 样例输入 5 7 样例输出 1 2 3 5 4 评测用例规模与约定 对于30%的评测用例,1 ≤ N, K ≤ 10, 1 ≤ w ≤ N, 1 ≤ s, c ≤ 30;
|
思路
首先如果以老师为单位考虑问题的话会变得很复杂,借钥匙时间、还钥匙时间、钥匙摆放顺序都比较难放在一起考虑,所以可以把借和还拆分成两个动作,根据题中要求,借动作的时间就是我们输入的借钥匙时间,还钥匙的时间是借钥匙的时间加上上课时长,再通过引入一个标志变量将借动作和还动作加以区分,再对存储了动作的数组进行排序,如果时间不一致则按照先后顺序就可以了,如果时间相同,则需要先执行还动作后执行借动作,如果动作也都是一致的,则需要根据动作执行的对象——钥匙的编号,从小到大执行,基于这种想法,建立了一个函数,再通过调用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;
}