编程语言
首页 > 编程语言> > 编程小记录——Leetcode 215

编程小记录——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