题目
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:
你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。
思路
该题的思路有很多,最简单的想法是将整个数组排序后直接输出第k大的数字,复杂度为O(nlogn)。
也可以用一个递减优先队列来做,当队列大小小于k时,直接将数字放进队列,队列大小等于k时,将队列头部数字pop掉。再将当前数字push。
第一种方法排序整个数组,这多做了很多无用功,因为我们只关心第k大个数,并不关心第k大后面的数的顺序如何,我们应该可以对排序方法进行优化以减少复杂度。
O(nlogn)的排序方法主要是快排和归并,归并应该是无法优化的了,在这里我们考虑快排,快排其实就是选一个数,将比它大的数放一边,比它小的放一边。而在这个题中,第k大的数要么是我们选的那个数,要么在选的那个数大的一边或者小的一边,O(nlogn)的快排是对这两边都排序的,但我们其实只需要排一边的序就可以了,因此这就达到了简化的目标。排序应该平均需要(n+n/2+n/4+……+1)次读取,复杂度为O(n)。
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
return getPos(nums , 0 , nums.size() , k);
}
int getPos(vector<int>& nums, int start , int end , int k){
int chooseNum = nums[start];
int low = start , high = end - 1 ;
while(low < high){
while(low < high && nums[high] <= chooseNum ){
high --;
}
nums[low] = nums[high];
while(low < high && nums[low] >= chooseNum ){
low ++;
}
nums[high] = nums[low];
}
nums[low] = chooseNum;
if(low == k - 1){
return chooseNum;
}
else if(low < k - 1){
return getPos(nums , low + 1 , end , k);
}
else{
return getPos(nums , start , low , k);
}
}
};