LeetCode—215.数组中的第K个最大元素(Kth Largest Element in an Array)——分析及代码（C++）

一、题目

在未排序的数组中找到第 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 ≤ 数组的长度。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-largest-element-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 排序

（1）思路

直接对数组排序，输出第 k 个元素。

（2）代码

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());
        return nums[nums.size() - k];        
    }
};

（3）结果

执行用时 ：12 ms, 在所有 C++ 提交中击败了 87.07% 的用户；
内存消耗 ：9.1 MB, 在所有 C++ 提交中击败了 89.75% 的用户。

2. 优先队列

（1）思路

设计一个升序排列的优先队列，并保持队列规模为 k（当规模大于 k 时 pop 出队首元素），则最终队首即第 k 个最大元素。

（2）代码

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int, vector<int>, greater<int>> pq;
        for (int i = 0; i < nums.size(); i++) {
            pq.push(nums[i]);
            if (pq.size() > k)
                pq.pop();
        }  
        return pq.top();
    }
};

（3）结果

执行用时 ：16 ms, 在所有 C++ 提交中击败了 66.47% 的用户；
内存消耗 ：9.2 MB, 在所有 C++ 提交中击败了 74.05% 的用户。

3. 快速排序（优化前）

（1）思路

参考快排思路，不断选取 key 并保持 key 左/右两侧元素均 小于/大于 key。
在本题中无需完成全部排序过程，当 key == nums[k] 时，即对应第 k 个最大元素。

（2）代码

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int left = 0, right = nums.size() - 1;
        int key_index = -1, posi = nums.size() - k;
        while (key_index != posi) {
            key_index = partition(nums, left, right);
            if (key_index < posi)//根据当前key的下标缩小区间
                left = key_index + 1;
            else if (key_index > posi)
                right = key_index - 1;
        }
        return nums[key_index];
    }

    int partition(vector<int>& nums, int left, int right) {
        int key = nums[left];
        int i = left, j = right;
        while (i != j) {//左右交替快速排序
            while (i != j && key < nums[j] + 1)
                j--;
            swap(nums[i], nums[j]);
            while (i != j && nums[i] < key + 1)
                i++;
            swap(nums[i], nums[j]);
        }
        return i;
    }
};

（3）结果

执行用时 ：28 ms, 在所有 C++ 提交中击败了 38.72% 的用户；
内存消耗 ：9.1 MB, 在所有 C++ 提交中击败了 88.21% 的用户。

4. 快速排序（优化后）

（1）思路

在上述思路基础上，结合二分法选取 key 增加随机性。

（2）代码

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int left = 0, right = nums.size() - 1;
        int key_index = -1, posi = nums.size() - k;
        while (key_index != posi) {
            key_index = partition(nums, left, right);
            if (key_index < posi)
                left = key_index + 1;
            else if (key_index > posi)
                right = key_index - 1;
        }
        return nums[key_index];
    }

    int partition(vector<int>& nums, int left, int right) {
        swap(nums[left], nums[(right + left) / 2])//结合二分法选取 key 增加随机性
        int key = nums[left];
        int i = left, j = right;
        while (i != j) {
            while (i != j && key < nums[j] + 1)
                j--;
            swap(nums[i], nums[j]);
            while (i != j && nums[i] < key + 1)
                i++;
            swap(nums[i], nums[j]);
        }
        return i;
    }
};

（3）结果

执行用时 ：8 ms, 在所有 C++ 提交中击败了 98.39% 的用户；
内存消耗 ：9.2 MB, 在所有 C++ 提交中击败了 84.90% 的用户。

三、其他

暂无。