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