Java 求数组中前k小的元素

    科技2026-02-20  6

    思路: 就是将数组里的元素先放进优先级队列(底层是个小堆)里,然后取优先级队列的前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的小根堆。

    Processed: 0.028, SQL: 12