技術標籤：菜鳥學Pythoncsp

CSP 201709-2 公共鑰匙盒 Python滿分程式碼

1. 題目

---

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

---

2. 解題思想

按存放鑰匙時間進行遍歷，先還鑰匙，再取鑰匙。

N,K = map(int,input().split())

#鑰匙盒
keys = [i+1 for i in range(N)]

#存放輸入資料
lis = []

#存放取鑰匙 和還鑰匙時間 用於後續遍歷
lis_L = []

for i in range(K):
    a,b,c = list(map(int,input().split()))
    lis.append([a,b,b+c])
    lis_L.append(b)
    lis_L.append(b+c)

# 存取時間列表轉集合，然後再轉列表 減少重複遍歷次數
lis_L = set(lis_L)
lis_L = list(lis_L)
lis_L.sort()

for i in lis_L:
    out = []
    # 先還鑰匙
    for j in range(K):
        if lis[j][2] == i:
            out.append(lis[j][0])
    # 多人還鑰匙時 按照從小到大存放
    if(len(out) > 0):
        out.sort()
        for k in range(len(out)):
            keys[keys.index(-1)] = out[k] # 空位置位置標記為 -1
                       
    # 取鑰匙
    for j in range(K):
        if lis[j][1] == i:
            keys[keys.index(lis[j][0])] = -1 # 取鑰匙後原位置標記為 -1 
keys = list(map(str,keys))
print(" ".join(keys))

3. 執行結果