Leetcode 215:寻找数组中的第 K
个大元素
基于Python的堆库 heapq
除了使用sorted
函数进行排序外,对于一个可迭代变量nums
和一个常数k
,可以使用heapq
找到前K
个最大值:
import heapq
return heapq.nlargest(k, nums)[-1]
同理可以找到第K
个最小值:
import heapq
return heapq.nsmallest(k, mums)[-1]