CCF201709-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 样例说明 　　第一位老师从时刻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。   思路：建立时间轴，将事件标记在时间轴上，同时利用好TreeMap与TreeSet的有序性来存放数据将大大简化操作

package publicKeyBox;

import java.util.TreeMap;
import java.util.TreeSet;
import java.util.Map;
import java.util.Scanner;

public class Main {

	static Map<Integer, TreeMap<Integer, TreeSet<Integer>>> timeBar = new TreeMap<>();// 时间轴
	@SuppressWarnings("resource")
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();// 挂钩数
		int k = sc.nextInt();// 上课老师数
		int[] keyBox = new int[N];// 钥匙盒
		// 将所有老师的借走与归还钥匙的事件，包括时间点与对应教室的钥匙记录在时间轴上
		for (int i = 0; i < k; i++) {
			int key = sc.nextInt();// 钥匙
			int bor = sc.nextInt();// 借走时间
			int ret = bor + sc.nextInt();// 归还时间
			createEvent(ret);
			timeBar.get(ret).get(0).add(key);// 登记归还信息进入时间轴
			createEvent(bor);
			timeBar.get(bor).get(1).add(key);// 登记借走信息进入时间轴
		}
		for (int i = 0; i < N; i++)
			// 初始化钥匙盒
			keyBox[i] = i + 1;

		for (TreeMap<Integer, TreeSet<Integer>> a : timeBar.values()) {
			int c = 0;// 记录空位寻址起点序号
			for (int b : a.get(0))// 先归还
			{
				for (; c < N; c++)// 找空位
				{
					if (keyBox[c] == 0) {
						keyBox[c] = b;
						break;
					}
				}
			}
			for (int d : a.get(1))// 后取出
			{
				for (int e = 0; e < N; e++) {
					if (keyBox[e] == d) {
						keyBox[e] = 0;// 空位记0
						break;
					}
				}
			}
		}

		for (int result : keyBox) {
			System.out.print(result + " ");// 输出keyBox
		}
	}

	public static void createEvent(int t) {
		if (!timeBar.containsKey(t))// 时间轴上时间点不存在
		{
			TreeMap<Integer, TreeSet<Integer>> baR = new TreeMap<>();// 该时间点存放钥匙总事件集合
			TreeSet<Integer> bor = new TreeSet<>();// 借钥匙事件集合
			TreeSet<Integer> ret = new TreeSet<>();// 还钥匙事件集合
			baR.put(0, ret);
			baR.put(1, bor);
			timeBar.put(t, baR);// 将事件标记在时间轴
		}
	}

}

---

方法更新：以上方法采用的是map的嵌套，下面方法解题思路相同，通过对象来实现。

package keyBox;

import java.util.*;

public class Main {
    static Map<Integer,Event> timeBar=new TreeMap<Integer,Event>();//利用TreeMap建立一条空时间轴
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N=sc.nextInt();//钥匙总数
        int K=sc.nextInt();//老师数
        int[] keybox = new int[N];//钥匙盒
        for (int i=0;i<N;i++)//对钥匙盒初始化
            keybox[i]=i+1;
        for(int i=0;i<K;i++)
        {
            int key=sc.nextInt();//老师使用的钥匙编号
            int startTime=sc.nextInt();//该钥匙借走时间
            int endTime=sc.nextInt()+startTime;//该钥匙归还时间

            if(!timeBar.containsKey(startTime))//判断时间轴上是否存在该事件发生的时间点
            {
                timeBar.put(startTime, new Event());//不存在则添加该事件发生的时间点
            }
            timeBar.get(startTime).addBor(key);//添加借钥匙事件

            if(!timeBar.containsKey(endTime))//判断时间轴上是否存在该事件发生的时间点
            {
                timeBar.put(endTime, new Event());//不存在则添加该事件发生的时间点
            }
            timeBar.get(endTime).addRet(key);//添加还钥匙事件

        }
        for (Event e: timeBar.values())//从0时刻开始遍历时间轴上所有有记录的时间点
        {
            int r=0;
            for (int k:e.ret)//优先归还，优先处理该时间点还钥匙事件
            for (;r<N;r++)
            {
                if(keybox[r]==0)//等于0即钥匙盒中该处无钥匙
                {
                    keybox[r] = k;//将钥匙挂于此空位
                    break;
                }
            }

            for (int k:e.bor)//处理该时间点借钥匙事件
                for (int i=0;i<N;i++)
                {
                    if(keybox[i]==k)//找到钥匙盒中对应的钥匙
                    {
                        keybox[i] = 0;//将钥匙取走，该处标记为0
                        break;
                    }
                }
        }
        for (int i:keybox)//输出最终钥匙盒中钥匙摆放情况
            System.out.print(i+" ");
    }
}
class Event//事件类
{
    TreeSet <Integer>bor=new TreeSet<Integer>();//借钥匙事件集
    TreeSet <Integer>ret=new TreeSet<Integer>();//还钥匙事件集
    public void addBor(int key)
    {
        bor.add(key);
    }
    public void addRet(int key)
    {
        ret.add(key);
    }
}