CCF-CSP：201709-2公共钥匙盒

CCF-CSP题目汇总

题目

编号： 201709-2

试题名称： 公共钥匙盒

时间限制： 1.0s

内存限制： 256.0MB

问题描述

  有一个学校的老师共用N个教室，按照规定，所有的钥匙都必须放在公共钥匙盒里，老师不能带钥匙回家。每次老师上课前，都从公共钥匙盒里找到自己上课的教室的钥匙去开门，上完课后，再将钥匙放回到钥匙盒中。
  钥匙盒一共有N个挂钩，从左到右排成一排，用来挂N个教室的钥匙。一串钥匙没有固定的悬挂位置，但钥匙上有标识，所以老师们不会弄混钥匙。
  每次取钥匙的时候，老师们都会找到自己所需要的钥匙将其取走，而不会移动其他钥匙。每次还钥匙的时候，还钥匙的老师会找到最左边的空的挂钩，将钥匙挂在这个挂钩上。如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还。如果同一时刻既有老师还钥匙又有老师取钥匙，则老师们会先将钥匙全还回去再取出。
  今天开始的时候钥匙是按编号从小到大的顺序放在钥匙盒里的。有K位老师要上课，给出每位老师所需要的钥匙、开始上课的时间和上课的时长，假设下课时间就是还钥匙时间，请问最终钥匙盒里面钥匙的顺序是怎样的？

输入格式

  输入的第一行包含两个整数N, K。
  接下来K行，每行三个整数w, s, c，分别表示一位老师要使用的钥匙编号、开始上课的时间和上课的时长。可能有多位老师使用同一把钥匙，但是老师使用钥匙的时间不会重叠。
  保证输入数据满足输入格式，你不用检查数据合法性。

输出格式

  输出一行，包含N个整数，相邻整数间用一个空格分隔，依次表示每个挂钩上挂的钥匙编号。

样例数据

样例输入1

5 2
4 3 3
2 2 7

样例输出1
1 4 3 2 5

样例说明1
  第一位老师从时刻3开始使用4号教室的钥匙，使用3单位时间，所以在时刻6还钥匙。第二位老师从时刻2开始使用钥匙，使用7单位时间，所以在时刻9还钥匙。
  每个关键时刻后的钥匙状态如下（X表示空）：
  时刻2后为1X345；
  时刻3后为1X3X5；
  时刻6后为143X5；
  时刻9后为14325。

样例输入2

5 7
1 1 14
3 3 12
1 15 12
2 7 20
3 18 12
4 21 19
5 30 9

样例输出2
1 2 3 5 4

评测用例规模与约定

  对于30%的评测用例，1 $\le$ N, K $\le$ 10, 1 $\le$ w $\le$ N, 1 $\le$ s, c $\le$ 30；
  对于60%的评测用例，1 $\le$ N, K $\le$ 50，1 $\le$ w $\le$ N，1 $\le$ s $\le$ 300，1 $\le$ c $\le$ 50；
  对于所有评测用例，1 $\le$ N, K $\le$ 1000，1 $\le$ w $\le$ N，1 $\le$ s $\le$ 10000，1 $\le$ c $\le$ 100。

解答

分析

  使用一个res数组来维护钥匙的在盒子中的存取、顺序状态。将老师上课的信息info存到数组或队列in中，然后按上课时间从小到大排序。接着判断每一个老师的上课信息，首先查看小根堆Q,表示目前正在上课的老师，且按下课时间从小到大排序，当下课时间相同的时候再按要是序号从小到大排。如果小根堆中有必当前上课老师更早下课的，就将他的钥匙放到res最左边的空位。然后去走当前上课老师需要的钥匙，res对应位置置-1表示空，然后将他所取的钥匙序号和下课时间存入结构体time，推入Q。
  当所有老师的信息都处理完以后，再查看Q是否还有没下课的老师，然后把他们的钥匙一一放回。

代码

#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

typedef struct{//存储老师上课的输入信息
	int w,s,c;
}info; 

typedef struct time{//存储老师下课时间和其正在使用的钥匙
	int w,c;
	bool operator < (const time &rhs) const {
		if(c == rhs.c)//如果下课时间相同就按照钥匙序号从小到大排序
			return w > rhs.w;
        return c > rhs.c;
    }
}time; 

bool cmp(info i1,info i2){//输入信息按上课时间从小到大排序
	return i1.s<i2.s; 
}

int main(int argc, char** argv) {
	int N,K;
	int res[1002];
	info in[1002];
	priority_queue<time> Q;
	
	cin>>N>>K;
	
	for(int i=1;i<=N;i++){
		res[i]=i;
	}
	
	for(int i=0;i<K;i++){
		cin>>in[i].w >>in[i].s >>in[i].c ;
	}
	
	sort(in,in+K,cmp);
	
	for(int i=0;i<K;i++){
		time t;
		t.w = in[i].w;
		t.c = in[i].c +in[i].s;
		
		while(!Q.empty() && Q.top().c<=in[i].s){//先将已经下课的老师的钥匙放回
			for(int j=1;j<=N;j++){
				if(res[j]==-1){
					res[j]=Q.top().w;
					break;
				}
			}
			Q.pop();
		}
		
		Q.push(t);//把当前老师的信息推入Q，表示正在上课
		
		for(int j=1;j<=N;j++){
			if(res[j]==in[i].w){
				res[j]=-1;
				break;
			}
		}
	}
	

	for(int j=1;j<=N;j++){//处理还没下课的老师并输出结果
		if(res[j]==-1){
			res[j]=Q.top().w;
			Q.pop();
		}
		cout<<res[j]<<" ";
	}
		
	return 0;
}