给你一亿个数据,从中找出前K个大的数据。
有两种解决办法。
1.直接将数据从大到小排序,然后取前K个。但是由于数据的数量过于庞大,要开辟很大的空间,很浪费内存,所以这种方法不建议使用。
2.用堆来解决。要找前K个大的数据,则将待找的元素的前K个元素建立大小为K的小根堆,小根堆的堆顶元素是这K个数据中最小的。从剩下的数据里取一个元素每次和堆顶元素比较,如果数组当中的当前元素大于堆顶元素,则将堆顶元素出堆,然后将当前元素入堆。再将其调整为小堆,如此再循环。
代码如下:
public static Integer[] findKNum(int[] array,int k){
//1.建立大小为K的堆
PriorityQueue<Integer> minHeap = new PriorityQueue<>
(k, new Comparator<Integer>() {
//调整的时候使用
@Override
public int compare(Integer o1, Integer o2) {
return o1.compareTo(o2);//建立小堆
//return o2.compareTo(o1);建立大堆
}
});
//2.
for(int i=0; i<array.length; i++){
if(minHeap.size()<k){
minHeap.add(array[i]);//前K个元素入堆
}else{
Integer top = minHeap.peek();
if(top!=null && top<array[i]){
minHeap.poll();
minHeap.add(array[i]);
}
}
}
Integer[] integers = new Integer[k];
for(int i=0; i<k; i++){
Integer top = minHeap.peek();
integers[i] = top;//从小到大输出
minHeap.poll();
}
return integers;
}