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

样例说明

把40元分成30元和10元，分别买3瓶和1瓶，其中3瓶送1瓶，共得到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.Scanner;

public class Main {
    public static void main(String args[]) {
        Scanner in = new Scanner(System.in);
        int keys = in.nextInt();// 钥匙的数量
        int teachers = in.nextInt();// 教师的数量
        int key[] = new int[keys + 1];// 模拟钥匙
        for (int i = 0; i <= keys; i++) {
            key[i] = i;// 元素值为钥匙的编号
        }

        int map[][] = new int[teachers][3];// 记录教师取钥匙的信息，map[i][1]为开始上课的时间，map[i][2]为上课的时长
        int start = 10000, end = 1;// start是最晚开始上课的时间，end是最早结束上课的时间
        for (int i = 0; i < teachers; i++) {// 输入数据，并找到最晚开始上课的时间start和最早结束上课的时间end
            for (int j = 0; j < 3; j++) {
                map[i][j] = in.nextInt();
            }

            // 输入完一行，处理一行，再输入另一行
            if (map[i][1] <= start) {// start存储最晚开始上课的时间
                start = map[i][1];// map[i][1]为开始上课的时间
            }

            map[i][2] += map[i][1];// 此时的map[i][2]是结束上课的时间

            if (map[i][2] >= end) {
                end = map[i][2];// end存储最早结束上课的时间
            }
        }

        int[] temp = new int[keys + 1];// 下标为钥匙编号，值为0时表示未借出，值为1时表示已借出
        for (; start <= end; start++) {// 最晚开始上课的时间小于等于最早结束上课的时间
            for (int i = 0; i <= keys; i++) {// 初始化钥匙状态
                temp[i] = 0;// 刚开始时所有钥匙都未借出
            }

            for (int i = 0; i < teachers; i++) {// 标记所有需要还的钥匙
                if (map[i][2] == start) {// 此时的map[i][2]为上课结束时间，start最开始为最晚开始上课时间，之后自增
                    temp[map[i][0]] = 1;// 标记需还钥匙，map[i][0]是钥匙的编号
                }
            }

            for (int i = 1; i <= keys; i++) {// 还钥匙，i为钥匙的编号
                if (temp[i] == 1) {// 编号为i的钥匙需要还
                    for (int j = 1; j <= keys; j++) {
                        if (key[j] == 0) {
                            key[j] = i;
                            break;
                        }
                    }
                }
            }

            for (int i = 0; i < teachers; i++) {// 借钥匙
                if (map[i][1] == start) {
                    for (int j = 1; j <= keys; j++) {
                        if (key[j] == map[i][0]) {
                            key[j] = 0;
                            break;
                        }
                    }
                }
            }
        }

        for (int i = 1; i <= keys; i++) {
            System.out.print(key[i] + " ");
        }
    }
}