Leetcode215. Kth Largest Element in an Array

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

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

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种利用分治法解决寻找数组中第K大元素的问题，并提供了两种解决方案：一种是通过桶排序的方法实现，适用于特定范围内的整数排序；另一种则是基于快速排序思想的分治法，通过不断划分数组直到找到目标元素。

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

这是一道分治的题目，去面网易的机器学习岗位的时候问了我这个问题，我那时候第一反应就是桶排序。。

先附上桶排序的代码如下，应该注意两个地方：1、排序的数有可能是负数；2、排序的数有可能是重复的，比如[99 99]，要找出第一大的数，这种情况对于判断第几大的时候要采用大于等于而不是直接用等于来判断。

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int array[20000];
        int i;
        for(i = 0; i < 20000; i++){
            array[i] = 0;
        }
        for(i = 0; i < nums.size(); i++){
            array[nums[i]+10000]++;
        }
        int count = 0;
        for(i = 19999; i >= 0; i--){
            if(array[i] >= 1){
                count = count + array[i];
            }
            if(count >= k)
                break;
        }
        return i-10000;
    }
};

接下来用分治的思想，跟快排差不多的思想，代码如下：

class Solution {
public:
    int partition(vector<int>& nums, int low, int high){
        int pivot = nums[low];
        while(low < high){
            while(low < high && nums[high] <= pivot) high--;
            nums[low] = nums[high];
            while(low < high && nums[low] >= pivot) low++;
            nums[high] = nums[low];
        }
        nums[low] = pivot;
        return low;
    }
    int findKthLargest(vector<int>& nums, int k) {
        int start = 0;
        int end = nums.size()-1;
        k--;
        while(1){
            int pivot = partition(nums, start, end);
            if (pivot == k){
                return nums[pivot];
            }
            else if(pivot < k){
                start = pivot+1;
            }
            else
                end = pivot-1;
        }
    }
};