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

样例输入

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。

解题思路

时间和钥匙都依附于教师存在,定义教师结构体,重写比较函数cmp,通过vector向量来形成一个有序的堆栈，按照题目要求的先后顺序将教师结构体进行排序。其中每个教师包含两种状态的时间点，1. 还钥匙时。2. 借钥匙时

具体代码以及详细注释

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int k, n, w , s, c, key[1001];

//定义依附于教师的结构体
struct Teacher{
    int keyNum; //所使用的钥匙编号 
    int time; //使用钥匙时开始或结束的时间点 
    int flag; // 在此时间点是还钥匙1还是借钥匙0 
}; 

int cmp(const Teacher &a, const Teacher &b){
    if(a.time != b.time){
        return a.time < b.time; //a的时间点小于b时，返回正值 
    }else if(a.flag != b.flag){// 先还后取 
        return a.flag > b.flag;
    }else{
        return a.keyNum < b.keyNum;
    }
} 
int main(int argc, char** argv) {
    std::ios::sync_with_stdio(false);
    cin >> n >> k;
    //定义Teacher向量，每一个老师都对应两个v,取钥匙的时候和还钥匙的时候为不同的向量 
    vector<Teacher> v;
    //初始化钥匙的位置 
    for(int i =1; i<=n; i++){
        key[i]=i;
    } 
    //向向量里面添加教师信息
    for(int i=0; i<k; i++){
        cin >> w >> s >> c;
        Teacher t;
        t.keyNum = w;
        t.time = s;
        t.flag = 0; //借钥匙
        v.push_back(t);
        t.keyNum = w;
        t.time = s+c;
        t.flag = 1; //还钥匙
        v.push_back(t); 
    } 
    //给向量按照重写的cmp函数进行排序
    sort(v.begin(), v.end(), cmp);
    for(int i=0; i<v.size(); i++){
        if(v[i].flag == 0){//借钥匙 
            for(int j=1; j<=n; j++){
                if(key[j] == v[i].keyNum){
                    key[j]=0;
                    break;
                }
            }
        }else{//归还钥匙 
            for(int k=1; k<=n; k++){
                if(key[k] == 0){
                    key[k]=v[i].keyNum;
                    break;
                }
            }
        }
    } 

    for(int i = 1; i<=n; i++){
        cout <<key[i] <<" ";
    } 
    cout << endl;
    return 0;
}

参考资料：http://blog.youkuaiyun.com/bodhiye/article/details/78276304?locationNum=7&fps=1