215. Kth Largest Element in an Array(快排的分治思想)

最新推荐文章于 2024-09-23 19:00:00 发布

原创最新推荐文章于 2024-09-23 19:00:00 发布 · 269 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种高效查找无序数组中第K大元素的方法，通过使用最小堆或快速选择算法实现。快速选择算法基于快速排序思想，通过递归分区来定位目标元素。

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

题意：

给你一个无序的数组，让你求第k大元素；

思路：

比较简单的是利用最小堆（优先队列）保存前k大的元素，最后返回堆顶即可。还有一种方法是利用快排分治的方法，判断每次确定的键值是不是所求的元素；

是的话结束递归返回答案，

不是的话：

1.键值是第x大元素（x>k），此时所求元素在键值的右边，继续在右边区间递归。

2.键值是第x大元素（x<k），此时所求元素在键值的左边，继续在左边区间递归（注意修改k的值，k的值应改为k-x）；

java代码：

class Solution {
    public int findKthLargest(int[] nums, int k) {
        return QuickSelect(nums,0,nums.length-1,k);
    }
    public int QuickSelect(int[] nums, int s, int e,int k){
        int temp = nums[s];
        int l = s;
        int r = e;
        while(l<r){
            while(l<r&&nums[r]>=temp){
                r--;
            }
            nums[l] = nums[r];
            while(l<r&&nums[l]<=temp){
                l++;
            }
            nums[r] = nums[l];
        }
        nums[l] = temp;
        if(e-l+1==k){
            return nums[l];
        }
        else if(e-l+1>k){
            return QuickSelect(nums,l+1,e,k);
        }
        else{
            return QuickSelect(nums,s,l-1,k-e+l-1);
        }
    }
}