[CSP] 201709-2 公共钥匙盒

题目 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

样例说明

第一位老师从时刻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 ≤ 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。

解题报告

思路

本题C++采用STL中的优先队列priority_queue解决，C的代码笔者尽可能模拟优先队列的数据操作过程。

此处将同一教室的借钥匙与还钥匙分开为两次操作act，以时间先后对全部操作diary[]进行排序，依据排序靠前的操作对钥匙盒key[]的状态进行模拟。

AC代码

C

#include <stdio.h>
#include <stdlib.h>
#include <string.h> //memset函数
#include <stdbool.h> //bool类型

typedef struct Act {
    int room; //教室编号
    bool state; //true为归还,false为取用
    int time; //动作发生时间
} act;

int cmp1(const void *a, const void *b) {
    //a、b经过强制类型转换后,变为结构体指针,而成员选择运算符(->)的优先级高于取值运算符(*),故采用括号让取值操作优先进行
    if (((act *) a)->time == ((act *) b)->time) {
        if (((act *) a)->state == ((act *) b)->state) {
            return ((act *) a)->room - ((act *) b)->room; //同一状态的两个操作,教室编号越小,优先级越高,升序
        } else return ((act *) b)->state - ((act *) a)->state; //同一时间的两个操作,还(true)比借(false)优先,降序
    } else return ((act *) a)->time - ((act *) b)->time; //时间越早(小),优先级越高,升序
}

int main() {
    int i, n, k, w, s, c, pos = 0; //pos为队首元素下标(或者理解为"第一个尚未处理的元素下标")
    scanf("%d %d", &n, &k);
    int key[n + 1];
    act diary[2 * k];
    for (i = 1; i <= n; i++) key[i] = i;
    for (i = 0; i < k; i++) {
        scanf("%d %d %d", &w, &s, &c);
        //C语言不允许此种结构体赋值方式:diary[pos] = {w, false, s}; 但允许以{}方式初始化某结构体
        act tmp1 = {w, false, s}; //借钥匙
        diary[pos] = tmp1;
        act tmp2 = {w, true, s + c}; //还钥匙
        diary[++pos] = tmp2;
        ++pos;
    }
    qsort(diary, 2 * k, sizeof(diary[0]), cmp1);
    act p;
    pos = 0;
    while (pos < 2 * k) {
        p = diary[pos++];
        if (p.state) {
            for (i = 1; i <= n; i++) {
                if (key[i] == 0) {
                    key[i] = p.room;
                    break;
                }
            }
        } else {
            for (i = 1; i <= n; i++) {
                if (key[i] == p.room) {
                    key[i] = 0;
                    break;
                }
            }
        }
    }
    for (i = 1; i <= n; i++) {
        printf("%d ", key[i]);
    }
    return 0;
}

时间使用:0ms 空间使用:2.675MB

C++

#include <iostream>
#include <queue>

using namespace std;

typedef struct Act {
    int room; //教室编号
    bool state; //true为归还,false为取用
    int time; //动作发生时间
} act;

struct tmp1 {
    bool operator()(act a, act b) {
        if (a.time == b.time) {
            if (a.state == b.state) {
                return a.room > b.room; //同一状态的两个操作,教室编号越小,优先级越高,小顶堆
            } else return a.state < b.state; //同一时间的两个操作,还(true)比借(false)优先,大顶堆
        } else return a.time > b.time; //时间越早(小),优先级越高,小顶堆
    }
};

int main() {
    priority_queue<act, vector<act>, tmp1> diary;
    int n, k, w, s, c;
    cin >> n >> k;
    int key[n + 1];
    for (int i = 1; i <= n; i++) key[i] = i;
    for (int i = 0; i < k; i++) {
        cin >> w >> s >> c;
        diary.push({w, false, s}); //借钥匙
        diary.push({w, true, s + c}); //还钥匙
    }
    act p;
    while (!diary.empty()) {
        p = diary.top();
        diary.pop();
        if (p.state) {
            for (int i = 1; i <= n; i++) {
                if (key[i] == 0) {
                    key[i] = p.room;
                    break;
                }
            }
        } else {
            for (int i = 1; i <= n; i++) {
                if (key[i] == p.room) {
                    key[i] = 0;
                    break;
                }
            }
        }
    }
    for (int i = 1; i <= n; i++) {
        cout << key[i] << " ";
    }
    return 0;
}

时间使用:15ms 空间使用:2.953MB