数组中的第k大数字

时间复杂度和空间复杂度击败了80%的人!

看到题目,有些同学就想问:哎!这道题不是很简单嘛,排序一下再数组寻找坐标不就完事儿了嘛

这么想就不对了哦,因为我们只需要求第k大的数字,而排序是把每个数字都排序,所以直接对数组进行排序的话时间复杂度就会很高

思路:这里用的是堆排序,首先把数组变成大顶堆数组(把数组看成一个堆,依次根据数组下标排成二叉树,再对二叉树进行大顶堆化,这样这个数组就是一个大顶堆啦),变成之后,数组第一个位置就是最大的位置,我们把这个位置和数组最后一个位置进行交换,再对第一个位置进行大顶堆化。

代码如下

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int len=nums.length;
        
        //先构造大顶堆
        for(int i=len/2;i>0;i--){
            heapSort(nums,i,len);
        }

        //找出第k大的数字
        for(int i=0;i<k;i++){
            if(i+1==k) return nums[0];
            int temp=nums[0];
            nums[0]=nums[len-1-i];
            nums[len-1-i]=temp;
            
            heapSort(nums,1,len-1-i);
        }

        return -1;
    }
    
    //这个函数的意思就是对二叉树中的位置i
    //依次和它的左孩子和右孩子比较,孩子大的话就和孩子交换顺序
    //交换完之后,还需要对这个位置递归,因为要保证每一个节点都是大顶堆
    //左孩子:2*i;右孩子:2*i+1(数组会-1)
    public void heapSort(int[] nums,int i,int n){
        int j=2*i;
        if(j<=n){
            if(j+1<=n){
                if(nums[j-1]<nums[j]) j+=1;
            }
            if(nums[j-1]>nums[i-1]){
                int temp=nums[j-1];
                nums[j-1]=nums[i-1];
                nums[i-1]=temp;
                heapSort(nums,j,n);
            }
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值