CSP 201709-2公共钥匙盒（Java）

问题描述

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

---

import java.util.*;
public class Main {
	/*
	 * BorrowKey类存储取钥匙的相关信息
	 */
	static class BorrowKey implements Comparable<BorrowKey> {
		int keyNum;	//钥匙编号
		int start;	//取走时间
		int time;	//时常
		public BorrowKey(int keyNum, int start, int time) {
			this.keyNum = keyNum;
			this.start = start;
			this.time = time;
		}
		public int compareTo(BorrowKey o) {		//根据取走的先后时间顺序排序
			return this.start-o.start;
		}
		
	}
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int K = sc.nextInt();
		int[] Key = new int[N+1];
		for(int i=1;i<=N;i++) {
			Key[i] = i;
		}
		int[][] flag = new int[N+1][2];
		BorrowKey[] borrowKey = new BorrowKey[K];
		for(int i=0;i<K;i++) {
			borrowKey[i] = new BorrowKey(sc.nextInt(), sc.nextInt(), sc.nextInt());
		}
		Arrays.sort(borrowKey);		//排序
		int count1 = 0;				//count1用来确定当前是否有空的挂钩
		int count2 = 0;				//标记当前是第count2次取走钥匙  (count2<K)
		int maxTime = 0;
		for(int i=1;;i++) {
			//还钥匙部分
			if(count1 != 0) {
				for(int j=1;j<=N;j++) {
					if(i == flag[j][1]) {
						for(int k=1;k<=N;k++) {
							if(Key[k]==0) {
								Key[k] = j;
								count1--;
								flag[j][0] = 0;
								flag[j][1] = 0;
								break;
							}
						}
					}
				}
			}
			//借钥匙部分
			if(count2 < K) {
				BorrowKey tmp = borrowKey[count2];
				if(tmp.start == i) {
					for(int j=1;j<=N;j++) {
						if(tmp.keyNum == Key[j]) {
							Key[j] = 0;
							count1++;
							flag[tmp.keyNum][0] = tmp.start;
							flag[tmp.keyNum][1] = tmp.start + tmp.time;
							if(flag[tmp.keyNum][1]>maxTime)
								maxTime = flag[tmp.keyNum][1];
							count2++;
							break;
						}
					}
				}
			}
			else {
				if(i>maxTime)
					break;
			}
		}
		for(int i=1;i<=N;i++)
			System.out.print(Key[i]+" ");
	}
}

这是一道模拟题，模拟老师取钥匙和归还钥匙的整个过程，可能是我想的过程太过于复杂了，没想到什么好的解决方式，最后运行时间超限了，如果是正式考试可能稍微好一点，因为正式考试给出的时间限制长一些，如果大家有更高好的方法欢迎评论。