CSP之公共鑰匙盒（模擬、排序、優先佇列）

問題描述

試題編號：	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。

 這道題目可以將每個時間點借還鑰匙的資訊儲存在結構體當中。每個結構體分別儲存鑰匙的編號、借還鑰匙的時間以及事件的標誌資訊（表示是借鑰匙還是還鑰匙）。

根據題目意思，先將每個結構體按照順序排好，然後模擬取鑰匙和還鑰匙的過程。當然在模擬的過程中需要用到優先佇列，優先佇列中存放的是鑰匙的位置資訊。

具體的C++程式碼如下

#include <stdio.h>
#include <algorithm>
#include <queue>
#include <cstring>
using namespace std;
struct Info
{
    int time;
    int flag;
    int key;
};

struct Info temp[2005];

bool cmp(struct Info a,struct Info b)
{
    if(a.time!=b.time) return a.time<b.time;
    else if(a.flag!=b.flag) return a.flag>b.flag;
    else return a.key<b.key;
}

int arr[1005];

int main()
{
    int N,K;
    int i,j;
    priority_queue<int,vector<int>,greater<int> >q;
    scanf("%d %d",&N,&K);
    for(i=1;i<=N;i++) arr[i]=i;
    int w,s,c;
    for(i=0;i<K;i++)
    {
        scanf("%d %d %d",&w,&s,&c);
        temp[i*2].time=s;
        temp[i*2].key=w;
        temp[i*2].flag=0;
        temp[i*2+1].time=s+c;
        temp[i*2+1].key=w;
        temp[i*2+1].flag=1;
    }
    sort(temp,temp+2*K,cmp);
    for(i=0;i<2*K;i++)
    {
        if(temp[i].flag==1)
        {
            int pos=q.top();
            arr[pos]=temp[i].key;
            q.pop();
        }
        else
        {
            for(j=1;j<=N;j++)
            {
                if(arr[j]==temp[i].key)
                {
                    q.push(j);
                    arr[j]=-1;
                    break;
                }
            }
        }
    }
    for(i=1;i<=N;i++)
    {
        if(i<N) printf("%d ",arr[i]);
        else printf("%d\n",arr[i]);
    }
    return 0;
}