寻找第 K 大的数

jiangjiang1011

于 2021-12-07 14:49:14 发布

阅读量1.1k

点赞数

分类专栏：后端开发_数据库文章标签：排序算法 leetcode 算法

本文链接：https://blog.csdn.net/jiangmin1994/article/details/121769053

版权

后端开发_数据库专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文探讨了三种不同的方法来找到数组中的第k大数：排序法（O(n*logn+k)），选择排序（O(n*k)），以及快速排序改进版（平均O(n)）。通过实例和代码展示了快速选择算法在partitioning过程中的应用。

摘要由CSDN通过智能技术生成

第 K 大的数

解法1:
可以对这个初识的数组按照从大到小先行排序，然后取出前k大，总的时间复杂度为O(n*logn + k)。
解法2：
可以进行选择排序或交互排序，K次选择后即可得到第k大的数。总的时间复杂度为O(n*k)
解法3:
利用快速排序的思想，从数组S中随机找出一个元素X，把数组分为两部分S1和S2。S1中的元素大于等于X，S2中元素小于X。这时有两种情况：

1. S1中元素的个数小于k，则S2中的第k-|S1|个元素即为第k大数；
2. S1中元素的个数大于等于k，则返回S1中的第k大数。时间复杂度近似为O(n)

class Solution {
    public int findKthLargest(int[] nums, int k) {
       int id = partition(nums, 0, nums.length-1, nums.length-k);
        return nums[id];
    }
    public int partition(int[] nums, int left, int right, int k){
        int i = left - 1;
        int pivot = nums[right];
        for(int j=left; j<=right; j++){
            if(nums[j]<pivot){
                i++;
                int temp = nums[i];
                nums[i] = nums[j];
                nums[j] = temp;
            }
        }
        i++;
        int temp = nums[i];
        nums[i] = nums[right];
        nums[right] = temp;
        if(i==k) return i;
        else if(i<k) return partition(nums, i+1, right, k);
        else return partition(nums, left, right-1, k);
    }
}