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

之前用JAVA写的在这里:https://www.cnblogs.com/syq816/p/9924711.html

用C++改良版的思路:用一个优先级队列来存放老师要进行的操作以及要操作的时刻,重载<运算符,使得时间早的排在前面,时间相同的取钥匙的排在前面,时间相同,操作也相同的,序号小的排在前面,最后遍历优先级队列完成操作。

改进的地方:之前是从开始时间到结束时间遍历时间轴,然后再看每个事件点有没有老师进行操作,同时操作的时候还要把同时操作的老师再存储起来再判断后进行操作,这无疑浪费了步骤和时间。而用队列实现之后极大的减少了遍历次数,提高了效率。

#include<iostream>
#include<queue>

using namespace std;

struct Teacher {
    int keyNum;//老师要借的钥匙编号
    int op;//老师的操作0还钥匙,1借钥匙
    int time;//老师进行操作时候的时间
    //先按照老师开始时间排序
    //如果时间一致,按操作顺序排序,先还再借
    //如果操作一样,根据钥匙编号从小到大排序
    bool operator <(Teacher t) const {
        if (time != t.time)
            return time > t.time;
        else if (op != t.op)
            return op < t.op;
        else
            return keyNum > t.keyNum;
    }
};

const int N = 1000;
int keys[N + 1];//钥匙盒

int main() {
    int n, k;
    cin >> n >> k;
    //初始化钥匙序号
    for (int i = 1; i <= n; i++) 
        keys[i] = i;

    priority_queue<Teacher> q;//优先级队列
    int w, s, c;
    Teacher t;
    //将每个老师要进行的取钥匙还钥匙的操作入队
    for (int i = 0; i < k; i++) {
        cin >> w >> s >> c;
        t.keyNum = w;//钥匙编号
        t.op = 0;//取钥匙
        t.time = s;//取钥匙的时间
        q.push(t);//入队
        t.op = 1;//还钥匙
        t.time = s + c;//还钥匙时间
        q.push(t);
    }
    //队列不为空时遍历队列完成接下来的操作
    while (!q.empty()) {
        t = q.top();
        q.pop();
        if (t.op == 0) {
            //取钥匙操作
            for (int i = 1; i <= n; i++) {
                if (t.keyNum == keys[i]) {
                    keys[i] = 0;//取走钥匙,当前的钥匙盒未空
                    break;
                }
                    
            }
        }
        else {
            //还钥匙操作
            for (int i = 1; i <= n; i++) {
                //找到第一个空盒子放进去
                if (keys[i] == 0) {
                    keys[i] = t.keyNum;
                    break;
                }
            }
        }
    }

    for (int i = 1; i <= n; i++) {
        if (i != 1)
            cout << " ";
        cout << keys[i];
    }
    cout << endl;
    system("pause");
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/syq816/p/12398871.html
今日推荐