Const
给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
nums
k
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。
输入: nums = [3,2,1,5,6,4], k = 2
nums = [3,2,1,5,6,4]
k = 2
输出: 5
5
输入: nums = [3,2,3,1,2,4,5,5,6], k = 4
nums = [3,2,3,1,2,4,5,5,6]
k = 4
输出: 4
4
1 <= k <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
要找第K大元素,我们可以维护一个具有K个元素的小根堆。 这样子按照小根堆定义,即堆顶元素为堆中最小元素。 我们不断往堆中插入元素,当堆中元素满了,我们判断假如的元素是否比堆顶元素要大。如果是就删除堆顶元素。然后将大数加入堆顶。 由于堆的特性,每次都会heapify,所以堆顶会不断放置新的最小元素。当所有元素放完了,在堆内部调整下,重新推到堆顶的元素就是第K大元素。 换句话说,小根堆不允许存在比堆顶小的元素,只要大就会往下沉。一轮下来,小的都被踢出去了,最后留下的堆顶第K个元素就是第K大元素。
215.数组中的第K个最大元素
给定整数数组
nums
和整数k
,请返回数组中第k
个最大的元素。请注意,你需要找的是数组排序后的第
k
个最大的元素,而不是第k
个不同的元素。你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。
示例 1:
输入:
nums = [3,2,1,5,6,4]
,k = 2
输出:
5
示例 2:
输入:
nums = [3,2,3,1,2,4,5,5,6]
,k = 4
输出:
4
提示:
1 <= k <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4