CCF——公共钥匙盒(2017-9)

问题描述
  有一个学校的老师共用N个教室,按照规定,所有的钥匙都必须放在公共钥匙盒里,老师不能带钥匙回家。每次老师上课前,都从公共钥匙盒里找到自己上课的教室的钥匙去开门,上完课后,再将钥匙放回到钥匙盒中。
  钥匙盒一共有N个挂钩,从左到右排成一排,用来挂N个教室的钥匙。一串钥匙没有固定的悬挂位置,但钥匙上有标识,所以老师们不会弄混钥匙。
  每次取钥匙的时候,老师们都会找到自己所需要的钥匙将其取走,而不会移动其他钥匙。每次还钥匙的时候,还钥匙的老师会找到最左边的空的挂钩,将钥匙挂在这个挂钩上。如果有多位老师还钥匙,则他们按钥匙编号从小到大的顺序还。如果同一时刻既有老师还钥匙又有老师取钥匙,则老师们会先将钥匙全还回去再取出。
  今天开始的时候钥匙是按编号从小到大的顺序放在钥匙盒里的。有K位老师要上课,给出每位老师所需要的钥匙、开始上课的时间和上课的时长,假设下课时间就是还钥匙时间,请问最终钥匙盒里面钥匙的顺序是怎样的?
输入格式
  输入的第一行包含两个整数NK
  接下来K行,每行三个整数wsc,分别表示一位老师要使用的钥匙编号、开始上课的时间和上课的时长。可能有多位老师使用同一把钥匙,但是老师使用钥匙的时间不会重叠。
  保证输入数据满足输入格式,你不用检查数据合法性。
输出格式
  输出一行,包含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 ≤ NK ≤ 10, 1 ≤ w ≤ N, 1 ≤ sc ≤ 30;
  对于60%的评测用例,1 ≤ NK ≤ 50,1 ≤ w ≤ N,1 ≤ s ≤ 300,1 ≤ c ≤ 50;
  对于所有评测用例,1 ≤ NK ≤ 1000,1 ≤ w ≤ N,1 ≤ s ≤ 10000,1 ≤ c ≤ 100。

 说明:本题重要的点有两个,一个是要设置一个计时器,计算最大的计时时间,减少循环的次数;一个是时间是一步一步加的,当同一个时间既有还又有取,先要还钥匙(归还钥匙的时候,如果有两个老师都要还,按钥匙编号排序,所以要把还钥匙的序列做一个排序,注意还钥匙是还到钥匙盒最左边的空闲位置),还了之后才能取钥匙。注意每一次计时,把归还钥匙的序列清空,因为都已经归还了。

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        int k = input.nextInt();

        int[][] teacher = new int[k][4]; // 初始化老师们的操作,多设置一列保存老师归还钥匙的时间
        for (int i = 0; i < k; i++) {
            for (int j = 0; j < 3; j++) {
                teacher[i][j] = input.nextInt();
            }
            teacher[i][3] = teacher[i][1] + teacher[i][2];
        }
    
        int[] key = new int[n]; // 初始化钥匙序列
        for (int i = 0; i < n; i++) {
            key[i] = i + 1;
        }

        int[] ret = new int[n]; // 钥匙归还序列,保存钥匙的编号 (钥匙编号从小到大)

        int time = 1; // 计时器

        int max = teacher[0][3]; // 最大计时时间

        for (int i = 0; i < k; i++) {
            if (max < teacher[i][3]) {
                max = teacher[i][3];
            }
        }

        while (time <= max) { // 计时启动
            // 先要归还钥匙
            // 初始化归还序列
            for (int i = 0; i < n; i++) {
                ret[i] = 0;
            }
            // 把所有要归还的钥匙放到归还序列
            int m = 0;
            for (int i = 0; i < k; i++) {
                if (teacher[i][3] == time) {
                    ret[m++] = teacher[i][0];
                }
            }
            // 有可能同一时刻,有多把钥匙要归还,按钥匙编号从小到大排序
            Arrays.sort(ret);
            // 归还钥匙,找到钥匙序列中,最左边空缺的位置(钥匙序列为0表示空缺位置)
            for (int i = 0; i < n; i++) {
                if (ret[i] != 0) {
                    for (int j = 0; j < n; j++) {
                        if (key[j] == 0) {
                            key[j] = ret[i];
                            ret[i] = 0;
                            break;
                        }
                    }
                }
            }

            // 然后再取钥匙
            for (int i = 0; i < k; i++) {
                if (teacher[i][1] == time) {
                    for (int j = 0; j < n; j++) { // 找对应钥匙在钥匙的位置
                        if (key[j] == teacher[i][0]) {
                            key[j] = 0;
                        }
                    }
                }
            }
            time++;
        }

        // 输出最终钥匙编号
        for (int x : key) { 
            System.out.print(x + " ");
        }
    }
}