LeetCode 215. 数组中的第K个最大元素

题目

在未排序的数组中找到第 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);
        }
    }
};