Python返回列表中的top5,时间复杂度O(1)

Python返回列表中的top5,时间复杂度O(1)

有同学可能想了,我用sort排序一下,再返回不就好了。
那有没有别的解决办法呢?

# 题目:长度10W的列表,返回top5的数。
# 要求:时间复杂度O(1)
#
# 解题思路:
# 用列表模拟栈,遍历列表;
# 定义一个长度为5的栈,取出前5个数字压栈
# 从第6个元素开始,用下一个数字和栈内最小的元素比较,如果大于,删除最小的元素,执行压栈;
# 返回列表,即为top5

def func1(list_raw):
    max_stack = list_raw[:5]
    for i in list_raw[5:]:
        if i > min(max_stack):
            max_stack.remove(min(max_stack))
            max_stack.append(i)
            
    return max_stack


import numpy as np

list1 = list(np.random.randint(1000, size=1000))
print(func1(list1))

猜你喜欢

转载自blog.csdn.net/qq_36963372/article/details/83898864