思路:把借和还当成独立事件,按题目所说顺序进行排序,然后按题意进行模拟即可。
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> pa;
const int N = 2005;
struct node
{
int id;
int t,flag; //flag为1表示借,为-1表示还
}a[N];
int ans[N];
bool cmp(node x,node y)
{
if(x.t==y.t)
{
if(x.flag==y.flag)
return x.id<y.id; //同一时间来还,按编号从小到大还
return x.flag<y.flag; //同一时间还比借优先
}
return x.t<y.t; //按时间排序
}
int main()
{
int n,k,num=0;
cin>>n>>k;
for(int i=1;i<=n;i++) ans[i]=i;
for(int i=0;i<k;i++)
{
int w,s,t;
cin>>w>>s>>t;
a[num++]=node({w,s,1});
a[num++]=node({w,s+t,-1});
}
sort(a,a+num,cmp);
for(int i=0;i<num;i++)
{
if(a[i].flag==1)
{
for(int j=1;j<=n;j++)
if(ans[j]==a[i].id)
{
ans[j]=-1;break; //表示该钥匙被借走
}
}
else
{
for(int j=1;j<=n;j++)
if(ans[j]==-1) //如果有空位可以挂钥匙
{
ans[j]=a[i].id;
break;
}
}
}
for(int i=1;i<=n;i++)
{
if(i>1) cout<<" ";
cout<<ans[i];
}
return 0;
}