CCF201709-02公共钥匙盒

自己写这题的时候有点静不下心，去找网上的代码看，看了之后更加静不下心了。于是决定静下心来自己写这个题，我的这个代码应该还是挺好懂的。

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

思路：

看上去好像有些长，实际上思路很清晰，建议先看思路再看代码
如果使用time++然后作判断的方法，当时间跨度较大的时候可能会比较费时。因此我自定义了一个节点类Time，此类中的time用于存储节点时刻，key用于存储该时刻要操作的钥匙，flag用于标识该钥匙是取出还是归还。每把钥匙取出的时候，它归还的时间已经决定了，因此在时间轴上会有2个时间节点。int box[10240]用于记录钥匙盒的使用情况，若某个位置为空，则数组对应位置存储-1，否则存储钥匙的编号

使用自定义的排序函数cmp(Time a,Time b)实现以下规则：

1. 若时间节点a,b的时刻time不相同，则直接对它们的time进行升序排序
2. 若时间节点a,b的时刻time相同，则分两种情况
   (1)如果a,b的操作是相同的，优先处理编号小的钥匙
   (2)如果操作不同，优先处理要归还的钥匙

对时刻表进行排序，排序后遍历每个时间节点，判断它们关联钥匙的存取情况，并遍历box模拟钥匙的存取操作(有比遍历钥匙盒更省时间的办法，比如使用链表来记录已经操作过的位置，直接遍历它们会比遍历整个钥匙盒更加快)
最后依次输出box数组的前N个值，即为答案
　　因为代码中有注释，所以长度稍微长了点

#include <iostream>
#include <algorithm>
using namespace std;
class Key{
public:
    int id;
    int start,end;
    Key(){id=start=end=0;}
    Key(int _id,int _start,int _end){
        id=_id;start=_start;end=_end;
    }
}key[1024];
class Time{
public:
    Key key;
    int time;
    int flag;//取为0，还为1
    Time():time(0),flag(0){}
}time[10240];

bool cmp(Time a,Time b)
{
    if(a.time==b.time){//时间节点相同
        if(a.flag==b.flag)return a.key.id<b.key.id;//存取相同时，优先处理编号小的钥匙
        else return a.flag>b.flag;//先还后取
    }
    return a.time<b.time;
}
int main()
{
    int cnt=0;//关键时间点计数器
    int box[1024]={0};
    int N,K;cin>>N>>K;
    for(int i=1;i<=N;i++){
        box[i]=i;//将钥匙挂在对应位置
    }
    for(int i=0;i<K;i++){
        int id,start,end;
        cin>>id>>start>>end;
        end+=start;//此时的end指持续时间，因此要加上开始时刻才是结束时刻

        Key *tmp=new Key(id,start,end);
        //将该钥匙的取出时间点加入时间轴
        time[cnt].key=*tmp;
        time[cnt].time=start;
        time[cnt++].flag=0;//flag=0为取出

        //将该钥匙的归还时间点加入时间轴
        time[cnt].key=*tmp;
        time[cnt].time=end;
        time[cnt++].flag=1;//flag=1为归还
    }
    sort(time,time+cnt,cmp);
    for(int i=0;i<cnt;i++){//对每个时间节点进行遍历
        Key key=time[i].key;
        if(time[i].flag==1){//还钥匙 
            for(int j=1;j<=N;j++){//寻找空的位置
                if(box[j]==-1){//该位置可用
                    box[j]=key.id;//将该位置挂上钥匙
                    break;
                }
            }
        }else{//取钥匙
            for(int j=1;j<=N;j++){//寻找钥匙所在的位置
                if(box[j]==key.id){
                    box[j]=-1;//将该位置置空
                }
            }
        }
    }
    for(int i=1;i<=N;i++){
        cout<<box[i]<<" ";
    }
    cout<<endl;
    system("pause");
    return 0;
}

总结：

如何将思路理清是关键，此外对自定义的sort比较函数需要分情况排序。