【常见面试算法】数组中第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;
    }
}
全部评论

相关推荐

叶扰云倾:进度更新,现在阿里云面完3面了,感觉3面答得还行,基本都答上了,自己熟悉的地方也说的比较细致,但感觉面试官有点心不在焉不知道是不是不想要我了,求阿里收留,我直接秒到岗当阿里孝子,学校那边的房子都退租了,下学期都不回学校,全职猛猛实习半年。这种条件还不诱人吗难道 然后现在约到了字节的一面和淘天的复活赛,外加猿辅导。华为笔试完没动静。 美团那边之前投了个base广州的,把我流程卡麻了,应该是不怎么招人,我直接简历挂了,现在进了一个正常的后端流程,还在筛选,不知道还有没有hc。
点赞 评论 收藏
分享
06-06 03:40
已编辑
电子科技大学 Java
在秋招的小白菜很想养修勾:一眼 苍穹外卖+谷粒商城,项目换一换吧,可以找一些付费知识星球博主带带,避免烂大街。多投投大厂,背背八股,你这学历乱杀了,等实习经验到位,到时候大厂闭眼选
投递美团等公司8个岗位
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务