CCF CSP 201709-2 公共钥匙盒

问题描述

试题编号：	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。

解题的代码如下：

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

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();//N个挂钩（教师）
		int k = scanner.nextInt();//老师上课
		
		int[][] teacher = new int[k][4];//记录每位教师的信息，最后一列用记录归还钥匙的时间
		
		int[] key = new int[n];
		
		for(int i=1;i<=n;i++){
			key[i-1]=i;//初始化钥匙盒(装编号)
		}
		
		for(int i=0;i<k;i++){
			for(int j=0;j<3;j++){
				teacher[i][j]=scanner.nextInt();
				teacher[i][3]=teacher[i][1]+teacher[i][2];//输入教师的信息，并计算归还钥匙的时间
				
			}
		}
		
		int time=1,max=0;//time为计时器
		
		for(int i=0;i<k;i++){
			if(teacher[i][3]>max)
				max=teacher[i][3];//求得max为最迟归还钥匙的时间
		}
		
		int[] ret = new int[k];//归还钥匙序列
		
		while(time<=max){
			
			for(int i=0;i<k;i++)
				ret[i]=0;//初始化归还序列,0为不需要归还，其它值为需要还
			
			int m=0;//作为归还序列的起始位置
			
			for(int i=0;i<k;i++){//遍历教师信息，如果有归还，则将钥匙序号加入归还序列ret中
				if(teacher[i][3]==time){
					ret[m++]=teacher[i][0];//将该时间需要归还的钥匙序号放在归还序列中
				}
			}
			//将归还序列从小到大排序
			Arrays.sort(ret);
			
			//进行归还
			for(int i=0;i<k;i++){
				if(ret[i]!=0){///如果该时间需要还的钥匙队列中的值不为0，则进行还钥匙操作
					for(int j=0;j<n;j++){
						if(key[j]==0){//从左到右，如果该钥匙位置为空，则归还
							key[j]=ret[i];
							ret[i]=0;//将已归还的钥匙清楚归还序列
							break;
						}
					}
				}
			}
			
			//再次遍历老师信息表，借钥匙
			for(int i=0;i<k;i++){
				if(teacher[i][1]==time){
					for(int j=0;j<n;j++){
						if(key[j]==teacher[i][0])//看一下钥匙在不在,不在为0，在就为当前位置的值
								key[j]=0;//钥匙可借则借走
					}
				}
			}
			time++;//下一时刻
		}
		for(int i=0;i<n;i++){
			System.out.print(key[i]+" ");
		}
		
	}

}