CCF真题-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。

样例输入

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。

解题思路

首先创建一个k*3的数组a，代表k个老师使用钥匙的记录。第一列是教室的序号，第二列是该记录借钥匙的时间，第三列是还钥匙的时间。然后对a进行排序，以借钥匙的时间为基准排序得到数组ab，以还钥匙的时间为基准排序得到数组ae。
遍历所有的时间（从1到可能最大的10101），先还钥匙后借钥匙，如果钥匙被借走了，res数组中该位置被置为-1。还钥匙的过程是：找到数组中第一个为-1的位置置为该钥匙的编号。
如果begin和end全部为k，说明所有借钥匙和还钥匙的流程记录全部都已完成，跳出循环。

代码

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

public class Main {
    static int res[];

    public static void main(String[] args) {
        Scanner src=new Scanner(System.in);
        int n=src.nextInt();
        int k=src.nextInt();
        int a[][] = new int[k][3];

        for(int i=0;i<a.length;i++){
            a[i][0]=src.nextInt();
            a[i][1]=src.nextInt();
            a[i][2]=src.nextInt()+a[i][1];
        }

        int ab[][] = a.clone();
        int ae[][] = a.clone();
        Sort(ab, 1);//对数组ab排序，以钥匙的开始使用时刻为标准进行排序
        Sort(ae, 2);//对数组ae排序，以钥匙的结束使用时刻为标准进行排序
        res = new int[n + 1];
        for (int i = 0; i < res.length; i++) {
            res[i] = i;
        }
        int time = 0, begin = 0, end = 0;
        while ((time++) < 10101) {
            //如果同一时刻既有老师还钥匙又有老师取钥匙，则老师们会先将钥匙全还回去再取出。
            //还钥匙
            for(;end<ae.length;end++){
                if(time<ae[end][2]) break;
                if(time==ae[end][2]){
                    for(int key=0;key<res.length;key++){
                        if(res[key] == -1){
                            res[key] = ae[end][0];
                            break;
                        }
                    }
                }
            }
            //取钥匙
            for(;begin<ab.length;begin++){
                if(time<ab[begin][1]) break;
                if(time==ab[begin][1]){
                    for (int key=0;key<res.length;key++) {
                        if(res[key] == ab[begin][0]){
                            res[key] = -1;
                            break;
                        }
                    }
                }
            }
            if(begin==ab.length&&end==ae.length) break;
        }

        for (int i = 1; i < res.length; i++) {
            System.out.print(res[i]+" ");
        }
        System.out.println();
    }

    private static void Sort(int[][] a,final int i){
        Arrays.sort(a, new Comparator<int[]>() {
            public int compare(int[] o1, int[] o2) {
                //如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还。所以在o1[i]相等的情况下以o1[0]为基准进行排序。
                if(o1[i]==o2[i]) return o1[0]-o2[0];
                return o1[i]-o2[i] ;
            }
        });
    }
}

结果