LeetCode题解–215. Kth Largest Element in an Array

最新推荐文章于 2023-12-15 19:39:48 发布

原创最新推荐文章于 2023-12-15 19:39:48 发布 · 291 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode题解专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种高效寻找无序数组中第k大元素的方法。通过改进快速排序算法，仅对部分数组进行排序来达到目的，显著提高了运行效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接

LeetCode题目：https://leetcode.com/problems/kth-largest-element-in-an-array/

难度：Medium

题目

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.
For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

从无序数组中找出第k大的数字。

分析

直接将整个数组做快速排序，时间复杂度是O(N*logN)，不能通过。
然后想到，找第k大数字其实并不需要对整个数组排序，每次排序后都可以根据k的大小删掉前一半或者后一半的数组，大大提高了程序运行的效率。

代码

class Solution {
public:
    void binary_find(vector<int> &nums, int left, int right) {
        int x = nums[left], i = left, j = right;
        while (i < j) {
            while (i < j && nums[j] < x) j--;
            while (i < j && nums[i] >= x) i++;
            if (i < j) swap(nums[i], nums[j]);
        }
        swap(nums[left], nums[i]);
        if (i == pos) ans = nums[i];
        else if (i < pos) binary_find(nums, i + 1, right);
        else binary_find(nums, left, i - 1);
    }

    int findKthLargest(vector<int> &nums, int k) {
        pos = k - 1;
        int len = (int) nums.size();
        binary_find(nums, 0, len - 1);
        return ans;
    }

private:
    int pos;
    int ans;
};