Leetcode 215. Kth Largest Element in an Array[medium]

最新推荐文章于 2025-07-29 14:24:32 发布

原创最新推荐文章于 2025-07-29 14:24:32 发布 · 316 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

Leetcode 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种在未排序数组中查找第K大元素的有效算法。通过类似快速排序的方法将数组分为两部分，并确定目标元素位于哪一部分。此外，还提供了一个简单的排序查找方法作为对比。

题目：
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.

Note:
You may assume k is always valid, 1 ≤ k ≤ array’s length.

课上的例题，用类似快排的方法将数组分成两个集合：小于等于x的集合和大于x的集合。这样就可以确定第k大的数字处于哪个集合了，而另一个集合就不需要再搜索（递归）了。
时间复杂度，期望O(n)。
这里写图片描述

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        return find(nums, 0, nums.size() - 1, nums.size() + 1 - k);
    }

    int find(vector<int>& v, int lt, int rt, int k) {
        if (lt == rt) return v[lt];
        random_shuffle(v.begin() + lt, v.begin() + rt);
        if (v[rt] < v[rt - 1]) swap(v[rt], v[rt - 1]);
        int a = lt, b = rt - 1;
        bool p = 1;
        while (a != b) {
            if (v[a] <= v[b]) {
                if (p == 1) a++;
                else b--;
            }
            else {
                swap(v[a], v[b]);
                if (p == 1) b--, p = 0;
                else a++, p = 1;
            }
        }
        if (k <= a - lt + 1) return find(v, lt, a, k);
        else return find(v, a + 1, rt, k - (a - lt + 1));
    }
};

当然更直观的方法还是排序直接找第k大。
很尴尬，O(nlogn)的方法反而快一些。
这里写图片描述

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