Java 求数组中前k小的元素

思路: 就是将数组里的元素先放进优先级队列(底层是个小堆)里,然后取优先级队列的前k个数就好了。
至于什么是优先级队列,看这里!
自己以前好笨啊!!!怎么都想不转这个问题!

    //求最小的k个数
    // (这种问题用优先级队列可简单了,不用纠结PriorityQueue为什么要这么用,因为它是定义优先级队列的,且它的底层是一个小堆,它就是那么个用法)
   	public int[] samlllestk(int[] arr,int k){
        if(arr==null||k<=0){
            return new int[0];
        }
        PriorityQueue<Integer> pq=new PriorityQueue<>();//创建一个优先级队列
        //将数组中的元素依次放进堆中
        for(int i=0;i<arr.length;i++){
            pq.offer(arr[i]);//当数组的元素放进来之后,就默认是一个小堆
        }
        //将优先级队列的前k个元素放进数组中
        int[] ret=new int[k];
        for(int j=0;j<k;j++){
            ret[j]=pq.poll();//移除最顶端元素后,队列内部自己调整,直到将前k个元素都移出来
        }
        return ret;
    }	

另一种思路:

前k个最小的元素:建立一个大小为k的大根堆。然后将后面的元素依次与堆顶元素相比较,比堆顶元素小的,就先将堆顶元素与最后一个元素交换,然后删除最后一个元素,再将剩余的元素调整为大根堆,将刚刚比堆顶元素小的元素,放进刚刚堆的最后面,再进行调整。

(另外一个想法:建立了有k个元素的大根堆后,若后面的元素比堆顶元素小,就用该元素直接覆盖堆顶元素,然后将此时的堆调整为一个大根堆,就这么一直比较,一直比较到最后一个元素,这时候,这个大根堆就是前k个最小的元素组成的)
前k个最大的元素:建立一个大小为k的小根堆。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值