1075 链表元素分类（25 分） python

最新推荐文章于 2022-12-09 19:18:01 发布

原创最新推荐文章于 2022-12-09 19:18:01 发布 · 779 阅读

0 ·

CC 4.0 BY-SA版权

PTA 专栏收录该内容

28 篇文章

订阅专栏

探讨在处理大规模数据集时，如何优化算法以避免超时错误。通过分析现有代码，提出可能的优化策略，包括数据结构选择、循环效率提升及算法复杂度降低等关键议题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最后两个测试点超时，各位老哥看看哪里可以优化~
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))