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))