【常见面试算法】数组中第k大的数


给定一个数组A,要求找到数组A中第K大的数字。对于这个问题,解决方案有不少,此处我只给出三种:

方法1:

对数组A进行排序,然后遍历一遍就可以找到第K大的数字。该方法的时间复杂度为O(N*logN)

方法2:

利用简单选择排序法的思想,每次通过比较选出最大的数字来,比较上K次就能找出第K大的数字来。该方法的时间复杂度为O(N*K),最坏情况下为O(N^2)。

方法3:

这种方法是本文谈论的重点,可以利用快排的思想,首先快排每次执行都能确定一个元素的最终的位置,如果这个位置是n-k(其中n是数组A的长度)的话,那么就相当于找到了第K大的元素。设确定的元素位置m的话,如果m > n - k大的话,那么第K大的数字一定在A[0]~A[m - 1]之间;如果m < n - k的话,那么第K大的数字一定在A[m+1]~A[n - 1]之间。
整个过程可以通过递归实现:
class Solution {
    
    static int kk; // 第k大元素的快排后索引位置;
    
    public int findKthLargest(int[] nums, int k) {
        kk = nums.length - k; 
        quickSort(nums, 0, nums.length-1);     
        return nums[kk];
    }
    
    private void quickSort(int[] data, int start, int end){
        if(start < end){
            int mid = partition(data, start,end);
            //当第k大位置的数字归位时,结束递归遍历;
            if(mid == kk) return;
            else if(mid > kk) quickSort(data, start, mid-1); 
            else quickSort(data, mid+1, end);
            
        }
    }
    
    private int partition(int[] data, int start, int end ){
        int key = data[start];
        if(start > end) return -1;
        while(start < end){
            //一定要从后往前遍历!
            while(start < end && data[end] >= key) end--;
            if(start < end) swap(data, start, end);
            while(start < end && data[start] <= key) start++;
            if(start < end) swap(data, start, end);            
        }
        
        return start;
    }
    
    private void swap(int[] data, int i, int j){
        int tmp = data[i];
        data[i] = data[j];
        data[j] = tmp;
    }
}
全部评论

相关推荐

ProMonkey2024:5个oc?厉害! 但是有一个小问题:谁问你了?😡我的意思是,谁在意?我告诉你,根本没人问你,在我们之中0人问了你,我把所有问你的人都请来 party 了,到场人数是0个人,誰问你了?WHO ASKED?谁问汝矣?誰があなたに聞きましたか?누가 물어봤어?我爬上了珠穆朗玛峰也没找到谁问你了,我刚刚潜入了世界上最大的射电望远镜也没开到那个问你的人的盒,在找到谁问你之前我连癌症的解药都发明了出来,我开了最大距离渲染也没找到谁问你了我活在这个被辐射蹂躏了多年的破碎世界的坟墓里目睹全球核战争把人类文明毁灭也没见到谁问你了(别的帖子偷来的,现学现卖😋)
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务