CCF 公共鑰匙盒（模擬、排序、優先佇列）
問題描述
| 試題編號： | 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;
}