编程小记录——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]
上一篇:215. 数组中的第K个最大元素


下一篇:2021-07-22