CSP 201709-2 【公共钥匙盒】

目录

一、问题描述

二、实现代码

总结

---

一、问题描述

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

二、实现代码

#include<iostream>
#include<queue>
#define INF 2147483647
using namespace std;
struct take{
	int time,key;
	take(int time = 0, int key = 0) : time(time), key(key){} 
	bool operator < (const take &x) const {//取钥匙看谁的时间小就可以了 
        return time > x.time;
    } 
};
struct return_{
	int time,key;
	return_(int time = 0, int key = 0) : time(time), key(key){} 
	bool operator < (const return_ &x) const {//优先级的比较（优先级小返回true) 
        if(time > x.time) return true;
        else if(time == x.time){
			return key > x.key;//如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还
		}
		return false;
    } 
};
priority_queue <take> t; 
priority_queue <return_> r; 
int N,K,a[1001];
int main()
{
	cin >> N >> K; 
	int start,last,i,key;
	for(i = 1;i <= N;i++) a[i]=i;
	for(i = 1;i <= K;i++){
		cin >> key >> start >>last;
		t.push(take(start,key));
		r.push(return_(start + last,key));
	}
	take tt; return_ tr;

	while( !t.empty() ){
		tt = t.top(); tr = r.top();
		if(tt.time < tr.time){//如果先拿 
			t.pop();
			for(i = 1;i <= N; i++){
				if(a[i] == tt.key){
					a[i] = 0;
					break;
				}
			}
		}
		else{   //如果同一时刻既有老师还钥匙又有老师取钥匙，则老师们会先将钥匙全还回去再取出
				//如果取钥匙的时间比还钥匙的时间大，则先还钥匙 
			r.pop();
			for(i = 1;i <= N; i++){
				if(!a[i]){
					a[i] = tr.key;
					break;
				}
			}
		}
		for(i = 1;i <= N; i++){
			if(!a[i]) cout << "X";
			else cout << a[i];
		}
		cout << endl;
	}
	while( !r.empty() ){//没有人取钥匙了，归还钥匙 
		tr = r.top(); r.pop();
		for(i = 1;i <= N; i++){
			if(!a[i]){
				a[i] = tr.key;
				break;
			}
		}
		for(i = 1;i <= N; i++){
			if(!a[i]) cout << "X";
			else cout << a[i];
		}
		cout << endl;
	}
	//for(i = 1; i <= N; i++) cout << a[i] <<" ";
}

---

总结

用数组a记录钥匙盒的状态，a[1]~a[n]为钥匙盒从左到右悬挂的钥匙编号。结构体take用来记录取走钥匙的时间和取走钥匙的编号，结构体return_用来记录还钥匙的时间和钥匙的编号，再以这两个结构体分别建立两个优先队列t和r。

对于t队列而言，时间越小则越先取钥匙，优先级越高；对于r队列而言，时间越小则越先还钥匙，若还钥匙的时间相同，则钥匙编号越小优先级越高。

输入数据时，取钥匙的时间为上课时间，将该节点加入t队列；还钥匙的时间为上课时间+上课的时长，将该节点加入r队列。

若t队列不为空，即还有老师需要取钥匙，则从t和r中队头元素，判断还钥匙在前还是取钥匙在前，若取钥匙在前则先取钥匙，若还钥匙在前（或还钥匙和取钥匙在同一时刻），则先还钥匙。重复执行该步骤，直到t为空。还钥匙操作：从1开始遍历到n，若a[i]为0（即该处可以悬挂钥匙），则将其放至a[i]处（即令a[i]=key）；取钥匙操作：从1开始遍历到n，若a[i]==key（即找到钥匙的所在处），令a[i]=0（即取走钥匙）。

再判断r队列是否为空，即是否有老师还未归还钥匙，若有，则进行归还操作，直至r为空。