LeetCode题解——Kth Largest Element in an Array

最新推荐文章于 2019-08-23 13:31:23 发布

susandebug

最新推荐文章于 2019-08-23 13:31:23 发布

阅读量673

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： Leetcode Kth Largest Array partition

本文链接：https://blog.youkuaiyun.com/u010025211/article/details/49448431

LeetCode 专栏收录该内容

100 篇文章

订阅专栏

本文介绍了一种高效算法，用于从无序数组中找到第K大的元素。通过使用快速排序的分区思想，实现了一个名为findKthLargest的函数，该函数能够避免完全排序数组，从而提高查找效率。

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

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.

class Solution {
public:
//方法一：建立一个堆维护k个最大的数,没有必要，因为只需要第k大，而不需要前k大的数
//方法二：partition and findKthLargest.//利用快排的思想
    int findKthLargest(vector<int>& nums, int k) {
        //if(k<1||k>nums.size()) return 0;
        int begin = 0, end = nums.size()-1;
        while(true){
            int index = partition(nums,begin,end);
            if(index==nums.size()-k) return nums[index];
            else if(index<nums.size()-k) begin = index+1;
            else end = index-1;
        }
    }
    int partition(vector<int>& nums,int begin,int end){
       // if(!nums.size() || begin<0 || end<0 || end>nums.size()-1) 
       //     throw("Invalid Parameters");
        if(end==begin) return begin;
        int idx = rand()%(end-begin)+begin;
        swap(&nums[idx], &nums[end]);
        int small = begin-1;
        for(idx = begin; idx<end; ++idx){
            if(nums[idx]<nums[end]){
                small++;
                if(small!=idx) swap(&nums[idx],&nums[small]);
            }
        }
        ++small;
        swap(&nums[small],&nums[end]);
        return small;
    }
    void swap(int*x,int*y)
    {
         int temp;
         temp=*x;
         *x=*y;
         *y=temp;
    }
};