leetcode0215. 数组中的第K个最大元素-medium

1 题目：数组中的第K个最大元素

官方标定难度：中

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

示例 1:

输入: [3,2,1,5,6,4], k = 2
输出: 5

示例 2:

输入: [3,2,3,1,2,4,5,5,6], k = 4
输出: 4

提示：

$1<=k<=nums.length<=10^5$
$-10^4<=nums[i]<=10^4$

2 solution

直接用堆排序，找到 top - k

代码

class Solution {
public:
int findKthLargest(vector<int> &nums, int k) {
    /*
     * 维护大小为 k 的堆
     * 时间复杂度为 (n + k)logk
     */
    priority_queue<int, vector<int>, greater<>> topK;

    for (int i: nums) {
        topK.push(i);
        if (topK.size() > k) topK.pop();
    }

    return topK.top();
}
};

结果

3 进阶

用快速排序思路，只找第 k 个元素，即不需要完全排序。

代码

class Solution {
public:

int partition(vector<int> &nums, int left, int right) {
    if (right <= left) return left;

    int pivot = nums[right];
    int i = left - 1;
    for (int j = left; j < right; j++) {
        if (nums[j] < pivot) {
            swap(nums[++i], nums[j]);
        }else if(nums[j] == pivot){
            if(rand() % 2){
                swap(nums[++i], nums[j]);
            }
        }
    }
    swap(nums[i + 1], nums[right]);
    return i + 1;
}


int quickSelect(vector<int> &nums, int left, int right, int index) {
    int pivotIndex = partition(nums, left, right);
    if (pivotIndex == index) return nums[index];
    else if (pivotIndex > index) return quickSelect(nums, left, pivotIndex - 1, index);
    return quickSelect(nums, pivotIndex + 1, right, index);
}

int findKthLargest(vector<int> &nums, int k) {
    /*
     * 用快速选择算法:
     * 即快速排序的思路
     */
    return quickSelect(nums, 0, nums.size() - 1, nums.size() - k);
}
};

结果