CCP-CSP认证考试 201709-2 公共钥匙盒 c/c++题解

题目描述

试题编号： 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。

题解：

这道题刚开始是很困扰我的，就是不会做，但是后面突然通了，请听我思路：

1. 把老师分成两半(有点吓人噢)，意思就是题目输入的是N个老师的取\还书编号w、取书时间s、还书时间s+c，所以就可以把老师分为取书老师和还书老师，记录下取书or还书的编号w、取书or还书的时间t、取书or还书的标志flag(flag=0还书、flag=1取书，这里这样定义是有用意的，下面会说的)
2. 排序：
   如果时间不等，按照时间从小到大排序。
   return A.t < B.t
   如果时间相等，还书的老师排在取书的老师前面return A.flag < B.flag（flag=0还书排前面，flag=1取数排后面）
   但是还要考虑：还书的老师可能在这一时刻有多个，就按照还书的编号从小到大排序 （取书的老师也可能在这里时刻有多个，也就是flag同时为1，但是取书的顺序无所谓，反正只要拿走对应的书即可）

 if(A.flag == B.flag && A.flag == 0)
{
       return A.w < B.w;
}

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>
#include <list>
#include <utility>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <bitset>
#include <iterator>
using namespace std;

typedef long long ll;
const int inf = 0x3f3f3f3f;
const ll  INF = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double E = exp(1.0);
const int MOD = 1e9+7;
const int MAX = 1e3+5;
int key[MAX];

struct Teacher// 包含了取钥匙的老师 和 还钥匙的老师
{
    int w;// 老师要取/还的钥匙编号
    int t;// 老师取/还钥匙的时间
    int flag;// 1-老师取钥匙 0-老师还钥匙
};
Teacher t[2*MAX];
int cmp(const Teacher A,const Teacher B)
{
    if(A.t != B.t)// 如果时间不同，就按时间从小到大排序
    {
        return A.t < B.t;
    }
    else
    {
        // 如果时间相同，就是所有时间相同的老师先还钥匙，再取钥匙
        // and 还钥匙的老师要按照编号从小到大排序
        if(A.flag == B.flag && A.flag == 0)
        {
            return A.w < B.w;
        }
        else
        {
            return A.flag < B.flag;
        }
    }
}


/*  规则：
        1.每次取钥匙的时候，找到自己所需要的钥匙将其取走。
        2.每次还钥匙的时候，挂在最左边的空的挂钩上。
        3.如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还。
        4.如果同一时刻既有老师还钥匙又有老师取钥匙，则老师们会先将钥匙全还回去再取出。
*/
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int n,k;
    while(cin >> n >> k)
    {
        // 初始化钥匙编号
        for(int i = 1; i <= n; i++)
        {
            key[i] = i;
        }
        int cnt = 0;// 取/还钥匙的老师数量
        int w,s,c;
        for(int i = 1; i <= k; i++)
        {
            cin >> w >> s >> c;
            t[cnt].w = w; t[cnt].t = s; t[cnt++].flag = 1;// 取钥匙的老师
            t[cnt].w = w; t[cnt].t = s+c; t[cnt++].flag = 0;// 还钥匙的老师
        }
        // 按照取/还钥匙的时间对老师进行排序
        // 如果时间相同，则还钥匙的排在前面
        sort(t,t+cnt,cmp);
        for(int i = 0; i < cnt; i++)
        {
            if(t[i].flag == 0)
            {
                // 如果是还钥匙，则找到key数组的第一个0放入钥匙即可
                for(int j = 1; j <= n; j++)
                {
                    if(key[j] == 0)
                    {
                        key[j] = t[i].w;
                        break;
                    }
                }
            }
            else
            {
                // 如果是取钥匙的话，则找到key数组对应的钥匙编号置0即可
                for(int j = 1; j <= n; j++)
                {
                    if(key[j] == t[i].w)
                    {
                        key[j] = 0;
                        break;
                    }
                }
            }
        }
        for(int i = 1; i <= n; i++)
        {
            cout << key[i] << " ";
        }
        cout << endl;
    }

    return 0;
}