[Leetcode]215. Kth Largest Element in an Array

最新推荐文章于 2022-05-24 22:39:06 发布

原创最新推荐文章于 2022-05-24 22:39:06 发布 · 155 阅读

CC 4.0 BY-SA版权

本文介绍了一种基于快速排序的partition思想来寻找数组中第K大的元素，通过避免完全排序，实现更高效的查找。首先介绍了暴力解法，即完全排序数组后返回第K大元素，但其复杂度较高。随后提出了一种更优解法，利用快速排序的partition过程，通过递归调用，直接定位到第K大元素的位置，大大降低了时间复杂度。

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

暴力解，复杂度是快排复杂度o(nlogn)，空间是o(1)

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

利用快排partition思想，得到更快的方法,使用《算法导论》上的快排算法，引入了多余的交换

class Solution {
public:
    int partition(vector<int>& nums,int k, int s, int t){
        int i = s - 1;
        for(int j = s; j < t; j++){
            if(nums[j] > nums[t]){
                i++;
                swap(nums[j],nums[i]);
            }
        }
        i++;
        swap(nums[i],nums[t]);
        return i;
    }
    int find(vector<int>& nums,int k, int s,int t){
        int q = partition(nums,k,s,t);
        if(q == k - 1) return nums[k-1];
        else if(q < k - 1){
            return find(nums,k,q+1,t);
        }else{
            return find(nums,k,s,q-1);
        }
    }
    int findKthLargest(vector<int>& nums, int k) {
        return find(nums,k,0,nums.size()-1);
    }
};