215. 数组中的第K个最大元素

传送门

代码

class Solution {
    public int findKthLargest(int[] nums, int k) {
        return quickSelect(nums,0,nums.length - 1,nums.length - k);
    }
    public int quickSelect(int[] a,int l,int r,int k) {
        if(l >= r) return a[l];
        int base = a[l + r >> 1];
        int i = l - 1,j = r + 1;
        while(i < j) {
            do i ++;while(a[i] < base);
            do j --;while(a[j] > base);
            if(i < j) swap(a,i,j);
        }
        if(k <= j) return quickSelect(a,l,j,k);
        else return quickSelect(a,j + 1,r,k);
    }
    public void swap(int[] a,int l,int r) {
        int t = a[l];
        a[l] = a[r];
        a[r] = t;
    }
}

思路

快速选择算法
时间复杂度 \(O(n)\)
参考快排的思路,但是每次只选择一边进行递归,不会把两边都递归
题目要求,第 k 大,也就是把数组排序后(下标从 0 开始),第 n - k 小的数
每次直接把数组以 base 为中心(这个 base 可以用 random 来取,可以防止被卡)切分成两半
然后把整个数组重新整理,拨开到两边,那么这样的话 base 所在的位置,就一定是排好序后,它下标的位置
最后 i j 退出循环的时候,一定在 base 的分界线处
所以,如果要求的下标 k 在左边,就继续递归左边\([l,j]\),否则就递归右边\([j + 1,r]\)
不断夹逼,就会找到在下标为 k 的数

posted @ 2020-12-13 14:00  南风--  阅读(104)  评论(0编辑  收藏  举报