CCF 201709-2 公共钥匙盒 Java

试题编号： 201709-2
试题名称： 公共钥匙盒
时间限制： 1.0s
内存限制： 256.0MB
问题描述：

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

一开始，把每个操作放在int[4]数组，放在ArrayList里，用Iterator遍历，然后时间也是从1开始+1，所以超时。

然后不用Iterator遍历，直接用for循环，优化了时间，取出不重复的时间数组排序，就不超时了。

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

public class gonggong_yaoshi {
    static ArrayList<int[]> act; 
    static int[] key;

    public static void main(String[] args) {
        new gonggong_yaoshi().run();
    }

    public void run() {
        Scanner sc = new Scanner(System.in);
        int keynum = sc.nextInt();
        int acts = sc.nextInt();

        key = new int[keynum];
        act = new ArrayList<>();

        for (int i = 0; i < key.length; i++)
            key[i] = (i+1);

        int[] time = new int[acts*2+1]; int k = 0, ti;
        while( acts-- > 0){
            int[] temp = new int[4];
            temp[0] = sc.nextInt();
            temp[1] = sc.nextInt();
            time[k++] = temp[1];
            temp[2] = sc.nextInt() + temp[1];
            time[k++] = temp[2];
            act.add(temp);
        }
        time[time.length-1] = Integer.MAX_VALUE;
        Arrays.sort(time);
        k = 0; acts = act.size();

        while( k < time.length-1 ){
            if( (ti = time[k++]) == time[k])
                continue;
            int[] re = new int[keynum];
            int[] bo = new int[keynum];
            int c1 = 0, c2 = 0;

            for( int i = 0; i < acts; i++){
                if( act.get(i)[2] == ti){
                    re[c1++] = act.get(i)[0];
                }
                if( act.get(i)[1] == ti)
                    bo[c2++] = act.get(i)[0];
            }
            if( c1 > 0 )
                retur(re, ti, c1);
            if( c2 > 0 )
                borrow( bo, c2 );
        }

        for (int is : key) 
            System.out.print(is + " ");
        sc.close();
    }

    public void retur(int[] re, int time , int c1) {
        Arrays.sort(re);
        for( int i = 0; i < re.length; i++){
            if( re[i] != 0){
                for( int j = 0; j < act.size(); j++){
                    if( act.get(j)[0] == re[i] && act.get(j)[2] == time && act.get(j)[3] == 0){
                        for( int i2 = 0; i2 < key.length; i2++){
                            if( key[i2] == -1){
                                key[i2] = re[i];
                                break;
                            }
                        }
                        act.get(j)[3] = 1;
                        break;
                    }
                }
            }
        }
    }

    public void borrow(int[] bo,int c2) {
        for( int i = 0; i < c2; i++){
            for( int j = 0; j < key.length; j++){
                if( key[j] == bo[i]){
                    key[j] = -1;
                    break;
                }
            }
        }
    }//end borrow
}