CCF--公共钥匙盒--优先队列模拟

问题描述

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

分析：按照题目要求，先根据时间进行排序再进行模拟即可，看代码吧：

import java.util.*;

public class Main {
    static Scanner in = new Scanner(System.in);
    static int  n,k;
    static int[] a = new int[1005];
    //按照借钥匙时间排序
    static Comparator<Node> com1 = new Comparator<Node>() {		
		@Override
		public int compare(Node o1, Node o2) {
			return o1.s-o2.s;
		}
	};
	//按照还钥匙时间排序
	static Comparator<Node> com2 = new Comparator<Node>() {		
		@Override
		public int compare(Node o1, Node o2) {
          return o1.e==o2.e?o1.num-o2.num:o1.e-o2.e;//题目规则
		}
	};
	//找到第一个-1就返回，题目说从左边开始还钥匙
    static int findPos(int []a,int k) {
    	for(int i = 1;i <= n;i++)
    		if(a[i]==k)
    			return i;
    	return 0;
    }
	public static void main(String[] args) {
		 n = in.nextInt();
		 k = in.nextInt();
		PriorityQueue<Node> q1 = new PriorityQueue<Node>(com1);
		PriorityQueue<Node> q2 = new PriorityQueue<Node>(com2);
		int num,s,e,eno,p;
		for(int i = 0;i < k;i++) {
	      num = in.nextInt();
			s = in.nextInt();
			e = in.nextInt() + s;
			Node no = new Node(num,s,e);
			q1.add(no);
			q2.add(no);
		}
	  for(int i = 1;i <= n;i++)
			a[i] = i;
	  while(q1.size()!=0&&q2.size()!=0) {
		//借的时间小于还的时间，说明此状态下钥匙是被借出的，用-1标记
		   if(q1.peek().s<q2.peek().e) {
				 p = findPos(a, q1.peek().num);
				 a[p] = -1;
				q1.poll();
		   }
		 //借的时间大于等于还的时间，说明此状态下钥匙是钥匙是被归还的的
		   else if(q1.peek().s>=q2.peek().e) {
				 eno = q2.peek().num;
				 p = findPos(a, -1);
				 a[p] = eno;
				 q2.poll();
			 }				
		}	
	  //借的时间肯定小于还的时间，那么对于剩下的没还的钥匙从左边一个一个挂上即可
	   while(q2.size()!=0) {
			eno = q2.peek().num;
			p = findPos(a, -1);
			 a[p] = eno;
			q2.poll();
		}
		for(int i = 1;i < n;i++)
		 System.out.print(a[i]+" ");
	   System.out.println(a[n]);
	}

}
class Node{
	int num,s,e;
	Node(int num,int s,int e){
		this.num = num;
		this.s = s;
		this.e = e;
	}
}