ccf 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。

---

  我的思路是 将老师取钥匙和放回钥匙的操作用优先级队列（这里需要重载“<”符号）来保存，优先级队列是基于结构体的，结构体中存放钥匙的编号以及本次操作的时间；对于取钥匙的老师来说，结构体中即为需要取的钥匙编号以及老师上课的时间；对于放回钥匙的老师来说，结构体中即为需要放回的钥匙编号以及老师下课的时间；然后在给定一个时钟计时器（time，初值为1），表示此时此刻的时间，每次循环，时钟计时器自增。
  当取钥匙的优先级队列中的首元素的时间 不等于 放回钥匙的优先级队列中的首元素的时间，即此时，只有一种操作，要么是放回要是，要么是取钥匙，并不冲突，只需注意，可能有该时刻是否有多次取或者放回操作；否则，此时，既有老师放回钥匙，也有老师取钥匙，那么我们就让老师按照要是的大小从小到大依次放回后，再其他老师取钥匙。

代码如下：

//公共钥匙盒
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
typedef struct op{
    int num;
    int time;
    op(int n, int t):num(n),time(t){};
    
    friend bool operator <(op p1, op p2){
        if(p1.time != p2.time){
            return p1.time > p2.time;
        }
        else{
            return p1.num > p2.num;
        }
    }
}op;

int main(){
    int k, n;
    cin >> k >> n;
    vector<int> keys;
    //初始化钥匙编号
    for(int i=1; i<=k; i++){
        keys.push_back(i);
    }
    priority_queue< op >take;//拿走钥匙
    priority_queue< op >put;//放回钥匙
    
    for(int i=0; i<n; i++){
        int key,start,time;
        cin >> key >> start >> time;
        op *o = new op(key, start);
        take.push(*o);
        o = new op(key, start+time);
        put.push(*o);
    }
    int time = 1;
    while(!put.empty()){
        if(take.top().time == put.top().time && time == take.top().time){
            //先放回
            while(!put.empty() && put.top().time == time){
                op tmp = put.top();
                put.pop();
                for(int i=0; i<k; i++){
                    if(keys[i] == 0){
                        keys[i] =tmp.num;
                        break;
                    }
                }
            }
            //再取出
            while(!take.empty() && take.top().time == time){
                op tmp = take.top();
                take.pop();
                for(int i=0; i<k; i++){
                    if(keys[i] == tmp.num){
                        keys[i] = 0;
                        break;
                    }
                }
            }
        }
        //取出放回时间不一致
        else{
            while(!take.empty() && take.top().time == time){
                op tmp = take.top();
                take.pop();
                for(int i=0; i<k; i++){
                    if(keys[i] == tmp.num){
                        keys[i] = 0;
                        break;
                    }
                }
            }
            while(!put.empty() && put.top().time == time){
                op tmp = put.top();
                put.pop();
                for(int i=0; i<k; i++){
                    if(keys[i] == 0){
                        keys[i] =tmp.num;
                        break;
                    }
                }
            }
        }
        time++;
    }
    
    cout << keys[0];
    for(int i=1; i<k; i++){
        cout <<" "<<keys[i];
    }
    return 0;
}

ღ( ´･ᴗ･` )