201709-2 公共钥匙盒

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。

思路就是将时刻排序出来，知道是借钥匙还是还钥匙，然后是先还在借，如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还。

#include<iostream>
#include<math.h>
#include<algorithm>
using namespace std;

// 还 
struct Return{
	int id; // 钥匙编号
	int end; // 下课时间
};

// 借 
struct Lend{
	int id; // 钥匙编号
	int start; // 上课时间
}; 

bool cmpL(Lend b1,Lend b2){
	return b1.start<b2.start;
}

bool cmpR(Return r1,Return r2){
	if(r1.end==r2.end) return r1.id<r2.id; // 同时还先还编号小
	else return r1.end<r2.end;
}

int main()
{
	// w, s, c，分别表示钥匙编号、开始上课的时间和上课的时长
    int n,k,w,s,c;
    cin>>n>>k;
    // 挂钩上的钥匙编号 
    int a[n+1]={0};
	Return r[k]; 
	Lend l[k];
	// 刚开始1..n 等于0表示被借走 
	for(int i=1;i<=n;i++){
		a[i]=i;
	}
	for(int i=0;i<k;i++){
		cin>>w>>s>>c;
		l[i].id=w;
		l[i].start=s;
		r[i].id=w;
		r[i].end=s+c;
	} 
	// 借 
	sort(l,l+k,cmpL);
	// 还 
	sort(r,r+k,cmpR);
	int index1=0,index2=0;
	while(index1!=k||index2!=k){
		// 还书 ，同时先还或者已经借完了
		if(r[index2].end<=l[index1].start||index1==k){
			for(int i=1;i<=n;i++){
				if(a[i]==0){ // 挂钩为空 
					a[i]=r[index2].id;// 将钥匙挂上去
					index2++;
					break;
				}
			}
		}else{// 借书 
			for(int i=1;i<=n;i++){
				if(a[i]==l[index1].id){//编号找钥匙 
					a[i]=0;
					break;
				}
			}
			index1++;
		} 
	}
	for(int i=1;i<=n;i++){
		cout<<a[i]<<" ";
	}
    return 0;
}