问题描述: | 问题描述 有一个学校的老师共用N个教室,按照规定,所有的钥匙都必须放在公共钥匙盒里,老师不能带钥匙回家。每次老师上课前,都从公共钥匙盒里找到自己上课的教室的钥匙去开门,上完课后,再将钥匙放回到钥匙盒中。 输入格式 输入的第一行包含两个整数N, K。 输出格式 输出一行,包含N个整数,相邻整数间用一个空格分隔,依次表示每个挂钩上挂的钥匙编号。 样例输入 5 2 样例输出 1 4 3 2 5 样例说明 第一位老师从时刻3开始使用4号教室的钥匙,使用3单位时间,所以在时刻6还钥匙。第二位老师从时刻2开始使用钥匙,使用7单位时间,所以在时刻9还钥匙。 样例输入 5 7 样例输出 1 2 3 5 4 评测用例规模与约定 对于30%的评测用例,1 ≤ N, K ≤ 10, 1 ≤ w ≤ N, 1 ≤ s, c ≤ 30;
思路:建立时间轴,将事件标记在时间轴上,同时利用好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);
}
}