问题描述
有一个学校的老师共用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。
个人做题总结:按照时间的发生顺序,用Set存储,然后先把此刻要还的钥匙全部还了,再把此刻要取的钥匙全部取了,再进入下一时刻。
import java.util.*;
public class Main{
private int keyNumber;
private int startTime;
private int endTime;
private int costTime;
private Main() {
}
private Main(int keyNumber, int startTime, int costTime) {
this.keyNumber=keyNumber;
this.startTime=startTime;
this.costTime=costTime;
}
public static void main(String[] args) {
new Main().run();
}
private void run() {
Scanner scan = new Scanner(System.in);
int N,K;
//N是挂钩个数,K是老师的个数
N = scan.nextInt();
K = scan.nextInt();
List<Main> teacherList = new ArrayList<>();
int[] keyStatus = new int[N];
for (int i=0; i < N; i++) {
keyStatus[i] = i+1;
}
for (int i=0; i < K; i++) {
//初始化每个对象,并存到集合
Main teacher = new Main(scan.nextInt(),scan.nextInt(),scan.nextInt());
teacherList.add(teacher);
}
Set<Integer> timeSet = new TreeSet<>();
for (Main teacher:teacherList
) {
teacher.endTime = teacher.startTime + teacher.costTime;
timeSet.add(teacher.startTime);
timeSet.add(teacher.endTime);
}
for (int i=0; i < teacherList.size()-1; i++) {
for (int i1 = i+1; i1 < teacherList.size(); i1++) {
if(teacherList.get(i).endTime == teacherList.get(i1).endTime){
if(teacherList.get(i).keyNumber > teacherList.get(i1).keyNumber){
Main temp = teacherList.get(i);
teacherList.set(i,teacherList.get(i1));
teacherList.set(i1,temp);
}
}
if(teacherList.get(i).startTime == teacherList.get(i1).endTime){
Main temp = teacherList.get(i);
teacherList.set(i,teacherList.get(i1));
teacherList.set(i1,temp);
}
}
}
//遍历有序的时间Set集合,根据时间发生的顺序去逐一更改挂钩的状态
for (Integer time:timeSet) {
// int minKeyNumber = teacherList.get(i).keyNumber;
Set<Integer> backKeyNumberSet = new TreeSet<>();
List<Integer> getKeyNumberList = new ArrayList<>();
//还回挂钩上的钥匙,遍历钥匙排序状态,如果有多位老师还钥匙,则他们按钥匙编号从小到大的顺序还。
backKey(teacherList, time,backKeyNumberSet, keyStatus);
//取钥匙
getKey(teacherList, time,getKeyNumberList, keyStatus);
//清空集合,供下时刻使用
backKeyNumberSet.clear();
getKeyNumberList.clear();
}
//输出钥匙的状态
for (Integer keyNumber:keyStatus
) {
System.out.print(keyNumber);
System.out.print(" ");
}
}
private void backKey(List<Main> teacherList,Integer time,Set<Integer> backKeyNumberSet,int[] keyStatus) {
//收集在此刻时间需要还回的钥匙,放入集合backKeyNumberSet
for (Main aTeacherList : teacherList) {
if (time == aTeacherList.endTime) {
backKeyNumberSet.add(aTeacherList.keyNumber);
}
}
//如果有需要还回的钥匙,按钥匙从小到大的编号从左往右的空挂钩挂回钥匙
if(backKeyNumberSet.size()>0) {
for (Integer backKeyNumber:backKeyNumberSet) {
for (int j=0; j < keyStatus.length; j++) {
if (keyStatus[j] == 0) {
keyStatus[j] = backKeyNumber;
break;
}
}
}
}
}
private void getKey(List<Main> teacherList,Integer time,List<Integer> getKeyNumberList,int[] keyStatus) {
//收集在此刻时间需要取的钥匙,放入集合getKeyNumberList
for (Main aTeacher : teacherList) {
if(time == aTeacher.startTime){
getKeyNumberList.add(aTeacher.keyNumber);
}
}
//如果此刻要取的钥匙集合不为空,则取出
if(getKeyNumberList.size()>0) {
for (Integer aGetKeyNumberList : getKeyNumberList) {
for (int j=0; j < keyStatus.length; j++) {
if (keyStatus[j] == aGetKeyNumberList) {
keyStatus[j] = 0;
break;
}
}
}
}
}
}