CSP公共钥匙盒python

20170902 公共钥匙盒

有一个学校的老师共用N个教室，按照规定，所有的钥匙都必须放在公共钥匙盒里，老师不能带钥匙回家。每次老师上课前，都从公共钥匙盒里找到自己上课的教室的钥匙去开门，上完课后，再将钥匙放回到钥匙盒中。
　　钥匙盒一共有N个挂钩，从左到右排成一排，用来挂N个教室的钥匙。一串钥匙没有固定的悬挂位置，但钥匙上有标识，所以老师们不会弄混钥匙。
　　每次取钥匙的时候，老师们都会找到自己所需要的钥匙将其取走，而不会移动其他钥匙。每次还钥匙的时候，还钥匙的老师会找到最左边的空的挂钩，将钥匙挂在这个挂钩上。如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还。如果同一时刻既有老师还钥匙又有老师取钥匙，则老师们会先将钥匙全还回去再取出。
　　今天开始的时候钥匙是按编号从小到大的顺序放在钥匙盒里的。有K位老师要上课，给出每位老师所需要的钥匙、开始上课的时间和上课的时长，假设下课时间就是还钥匙时间，请问最终钥匙盒里面钥匙的顺序是怎样的？

样例输入
5 2
4 3 3
2 2 7
样例输出
1 4 3 2 5

样例输入
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。

不设最大结束时间时，30分
超时代码`.

// 八十分，超时
n, k = map(int, input().split())
lst = []
for i in range(k):
    lst.append(list(map(int, input().split())))
for i in range(k):
    lst[i][2] = lst[i][1] + lst[i][2]
key = [i+1 for i in range(n)]
end_time = [i[2] for i in lst]
for i in range(1, max(end_time) + 1):
    back = []
    borrow = []
    for teacher in lst:
        if teacher[2] == i:
            back.append(teacher)
        if teacher[1] == i:
            borrow.append(teacher)
    for t in sorted(back, key=lambda x: x[0]):
        index = key.index(0)
        key[index] = t[0]
        # print(i, key)
    for t in borrow:
        key[key.index(t[0])] = 0
        # print(i, key)

for i in key:
    print(i, end=' ')

//如果进入时间循环后不遍历全部选择，而是只挑出相应的借出和归还时间进行遍历，100
n, k = map(int, input().split())
lst = []
for i in range(k):
    lst.append(list(map(int, input().split())))
for i in range(k):
    lst[i][2] = lst[i][1] + lst[i][2]
key = [i+1 for i in range(n)]
end_time = [i[2] for i in lst]
start_time = [i[1] for i in lst]

for i in range(1, max(end_time) + 1):
    if i in end_time:  # 钥匙数量少于时间长度，选取结束时间做循环体
        back = []
        for j in range(len(end_time)):
            if end_time[j] == i:
                back.append(lst[j][0])
        for back_key in sorted(back):
            key[key.index(0)] = back_key

    if i in start_time:
        for j in range(len(start_time)):
            if start_time[j] == i:
                key[key.index(lst[j][0])] = 0

for i in key:
    print(i, end=' ')