1075 链表元素分类 (25 分) python

最后两个测试点超时,各位老哥看看哪里可以优化~
f_address,N,K = input().split(" ")
K = int(K)
N = int(N)
all_list = []
for i in range(N):
    ipt_list = input().split(" ")
    ipt_list[1] = int(ipt_list[1])
    all_list.append(ipt_list)
#all_list = [['23333', '10', '27777'], ['00000', '0', '99999'], ['00100', '18', '12309'], ['68237', '-6', '23333'], ['33218', '-4', '00000'], ['48652', '-2', '-1'], ['99999', '5', '68237'], ['27777', '11', '48652'], ['12309', '7', '33218']]
all_list1 = []
all_list2 = []
all_list3 = []
while 1:
    if all_list == []:
        break
    for i in all_list:
        if i[0] == f_address:
            if i[1]<0:
                i[1] = str(i[1])
                all_list1.append(i)
            elif 0<=i[1]<=K:
                i[1] = str(i[1])
                all_list2.append(i)
            else:
                i[1] = str(i[1])
                all_list3.append(i)
            f_address = i[2]
            all_list.remove(i)
sort_all_list = all_list1+all_list2+all_list3
sort_all_list[N-1][2] = "-1"
a = sort_all_list[N-1][0]
for i in range(N-2,-1,-1):
   sort_all_list[i][2] = a
   a = sort_all_list[i][0]
for i in sort_all_list:
    print(" ".join(i))

猜你喜欢

转载自blog.csdn.net/weixin_41775301/article/details/87890093