[Python]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

#公共钥匙盒

N,K = map(int,input().split())
a = []
b = []
timeline = {}
ad = {}
for i in range(K):
    a.append(list(map(int,input().split())))
    if a[i][1] not in b:
        b.append(a[i][1])
    if a[i][1]+a[i][2] not in b:
        b.append(a[i][1]+a[i][2])
b.sort()
for i in b:
    timeline[i]  = []
for i in a:
    timeline[i[1]].append(i[0])
    timeline[i[1]+i[2]].append(-i[0])
for i in range(N):
    ad[i+1] = i+1
for i in timeline:
    re = []
    borr = []
    for j in timeline[i]:
        if j < 0:
            re.append(j)
        else:
            borr.append(j)
    re.sort(reverse=True)
    borr.sort()
    if len(re):
        for l in re:
            for m in ad:
                if ad[m] == 0:
                    ad[m] = -l
                    break
    if len(borr):
        for n in borr:
            for o in ad:
                if ad[o] == n:
                    ad[o] = 0
                    break
for i in ad:
    print(ad[i],end=' ')

思路

  1. 用字典timeline来存储每个时间点的钥匙借还状态,钥匙编码为正数说明为借,钥匙编码为负数说明为还。
  2. 用字典ad来存储钥匙框上的位置,初始时为{1:1,2:2,3:3,4:4……n:n},当借走钥匙时,相应位置的值变为0,还钥匙时从最左边的0的位置处开始还。
  3. 遍历timeline,对每个时间点对应的钥匙序列,用两个数组进行排序,其中一个数组re[]储存归还的钥匙,里面的值均为负数,按从大到小排列;另一个数组borr[]储存要借的钥匙,里面的值为正数,按从小到大排列(不影响)。然后将钥匙先归还:归还到位置的值为0的地方;再对钥匙借:查找要借的钥匙的位置,然后将其值置为0。
  4. 输出最后的ad字典的值即可。

猜你喜欢

转载自www.cnblogs.com/SavvyM/p/11784658.html