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。
一、解题思路
一开始使用的是时刻+1这种循环方式，在每一时刻判断取钥匙还是还钥匙，所有钥匙还完循环结束。虽然可以得到正确的钥匙编号排放信息，但是运行超时了。因此，我又转换了思路，按照开始时间，结束时间进行判断。

我们可以分别使用两个结构体存储拿钥匙(be)和还钥匙(en)的信息，be.num表示拿走的钥匙编号，be.time表示拿走钥匙的时间，en.num表示归还的钥匙编号，en.time表示归还钥匙的时间。

对be，en，排序分别按照time升序排序，如果time相等，按照num升序排序。（题目中说明如果同时有多把钥匙待归还，按照编号由小到大归还，所以需要进行排序）

如果be.time小于en.time，则拿走编号为be.num的钥匙，如果be.time大于或等于en.time，则归还编号为en.num的钥匙。（题目中说明同时有老师取钥匙和还钥匙，先把钥匙还完再取钥匙)

使用循环时，注意循环因子的大致范围，注意运行超时问题。
二、代码

#include <iostream>
#include<algorithm>
using namespace std;
struct Key
{
    int num;
    int time;
};
bool cmp(Key x,Key y)
{
    if(x.time!=y.time)
        return x.time<y.time;
    if(x.num!=y.num)
        return x.num<y.num;//如果time相等按照num升序排序
}
int fin(int key[],int n,int t)//用于返回指定元素t在数组中的下标
{
    for(int i=0;i<n;i++)
    {
        if(key[i]==t)
        {
            return i;
            break;
        }
    }
}
int main()
{
    int n,k;
    cin>>n>>k;
    int key[n];
    Key be[k];
    Key en[k];
    for(int i=0;i<n;i++)
        key[i]=i+1;;
    for(int i=0;i<k;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        be[i].num=a;
        be[i].time=b;
        en[i].num=a;
        en[i].time=b+c;
    }
    sort(be,be+k,cmp);
    sort(en,en+k,cmp);
    int p=0;
    int t=0;
    for(int i=0;i<k;)
    {
        for(int j=t;j<k;)
        {
            if(be[j].time<en[i].time)
            {
                t++;//用来表示be.time>=en.time的下标
                key[fin(key,n,be[j].num)]=-1;
                j++;//拿走一个钥匙下标加 1，归还钥匙时be下标不动，还完再拿，所以j++不放在for循环里
            }
            else if(be[j].time>en[i].time||be[j].time==en[i].time)
            {
                key[fin(key,n,-1)]=en[i].num;//归还钥匙则i++
                i++;
            }
        }//归还钥匙有两种情况所以i也不放在for循环里
        if(t==k)
        {
            key[fin(key,n,-1)]=en[i].num;//归还钥匙则i++
            i++;
        }
    }
    for(int i=0;i<n;i++)
        cout<<key[i]<<" ";
    return 0;
}