编程小记录——Leetcode 215
作者:互联网
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]
标签:heapq,215,return,nums,编程,import,Leetcode 来源: https://blog.csdn.net/bengyanluo1542/article/details/121367957