python ccf题解 201709-2 公共钥匙盒

问题描述

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

代码

#公共钥匙盒
n,k = map(int,input().split())
taket = {}
returnt = {}
timeline = {}
for i in range(k):
    w,s,c = map(int,input().split())
    if(s not in timeline):#借钥匙
        timeline[s]=[]
    timeline[s] += [w]
    timeline[s].sort()
    if(s+c not in timeline):#还钥匙
        timeline[s+c]=[]
    timeline[s+c] += [-(n-w)]#方便排序,从小到大归还
    timeline[s+c].sort()

box = list(range(1,n+1))

for t in sorted(timeline.keys()):
    for x in timeline[t]:
        if(x <= 0):#还钥匙
            box[box.index(0)] = n+x
        else:#借钥匙
            box[box.index(x)] = 0
print(" ".join(map(str,box)))       

整体思路：

首先定义一个字典timeline存储借还钥匙的时间点，第一个循环中读取输入数据，借钥匙时间为s，还钥匙时间为s+c，时间点不在timeline的键中就将该时间点的值初始化为一个列表，列表中存储该时间点发生的所有借还事件，借钥匙为正值w，还钥匙为负值-(n-w)，（因为还钥匙要从左边开始还，所以应该先遍历编号小的位置，w小则-(n-w)也小，或者也可以用其他形式表示还钥匙，方便表示就行），时间点在timeline的键中就加入借还事件到对应值的列表中并排序，还钥匙负值、借钥匙为正值表示先还后借。

然后声明一个列表box序号表示挂钩编号，值表示挂的钥匙的编号，开始时钥匙编号从1到N排列。

第二个循环根据时间从小到大遍历timeline，对于每个时间点t，timeline[t]中存储该时间点借还事件列表，列表中的值从小到大排序。先遍历到负值为还钥匙，box[box.index(0)]找到序号最小的的值为0的项（表示最左边空的挂钩），赋值为n+x即(n+(-(n-w)))为w，表示归还到挂钩上钥匙的编号；还完钥匙借钥匙，box[box.index(w)]为要借钥匙的位置，赋值为0表示取走钥匙。

这样一遍遍历相当于模拟了借还钥匙的过程，最后box中的编号就是相应钥匙的位置。