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。

//
// Created by asimov on 2020/3/24.
//

#ifndef CSP_201709_2_H
#define CSP_201709_2_H

#endif //CSP_201709_2_H

#include "bits/stdc++.h"

using namespace std;

struct teacher {
    int id;
    int start;
    int length;
    int end;

    teacher(int _id, int _start, int _length, int _end) : id(_id), start(_start), length(_length), end(_end) {}
};

vector<teacher> teachers;
vector<int> keys;
int N, K;//N把钥匙　K个老师
int maxLen = 0;//模拟时间
vector<teacher> returnKey;//该时刻还钥匙的老师
bool cmp(teacher a, teacher b) {
    return a.id < b.id;
}

int fun() {
    cin >> N >> K;
    //初始化钥匙
    for (int i = 0; i < N; i++) {
        keys.push_back(i + 1);
    }


    for (int i = 0; i < K; i++) {
        int id, start, length;
        cin >> id >> start >> length;
        if (start + length > maxLen)
            maxLen = start + length;
        teachers.push_back(teacher(id, start, length, (start + length)));
    }

//模拟取钥匙
    for (int i = 1; i <= maxLen; i++) {

        returnKey.clear();
        //寻找当前时刻还钥匙的老师
        for (int j = 0; j < teachers.size(); j++) {
            if (i == teachers[j].end) {
                //到了还钥匙的时间
                returnKey.push_back(teachers[j]);
                //   cout<<"and"<<teachers[j].id<<endl;
            }
        }

        //还钥匙　从小到大的顺序放到最左边
        if (returnKey.size() > 1) {
          //  cout << "有" << returnKey.size() << "人还钥匙" << endl;
            //有多人同时还
            sort(returnKey.begin(), returnKey.end(), cmp);
            for (int m = 0; m < returnKey.size(); m++) {
                for (int mkey = 0; mkey < N; mkey++) {
                    //id为0表示位置空缺

                    if (keys[mkey] == 0) {
                    //    cout << returnKey[m].id << "来还钥匙" << endl;
                        keys[mkey] = returnKey[m].id;


                        break;
                    }

                }
            }
        } else if (returnKey.size() == 1) //只有一人还
        {

           // cout << "只有一人还钥匙" << endl;
            for (int mkey2 = 0; mkey2 < N; mkey2++) {
                //id为0表示位置空缺
                if (keys[mkey2] == 0) {
                    //cout << returnKey[0].id << "来还钥匙" << endl;
                    keys[mkey2] = returnKey[0].id;

                    break;
                }


            }

        }

        //借钥匙
        for (int j = 0; j < teachers.size(); j++) {
            if (i == teachers[j].start) {
               // cout << teachers[j].id << "借钥匙" << endl;
                for (int i = 0; i < N; i++) {
                    if(keys[i]==teachers[j].id )
                        keys[i]=0;
                }

            }

        }

    }

    for (int i = 0; i < N; i++) {
        cout << keys[i] << " ";
    }

    return 0;
}

/*
5 2
4 3 3
2 2 7


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

 */