题目描述:
从一组数组中选取最大(或最小) K 个元素
思路:
- 建立一个大小为 k 的小顶堆(如果求最小k对则建大顶堆)
- 把数放到优先队列里,堆顶为堆中最小数
- 若取下一个数比较 比堆顶大则入堆 这样结束后堆中即是最大k个数
代码实现:
import java.util.*;
public class TestTOPk {
public static void topK(int[] array,int k){
//建立大小为k的小堆
PriorityQueue<Integer> minHeap = new PriorityQueue<>(k,new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o1 - o2;
}
});
//遍历数组
for (int i = 0; i < array.length; i++) {
if (minHeap.size() < k){
minHeap.offer(array[i]);
}else{
int top = minHeap.peek();
if(array[i] > top){
minHeap.poll();
minHeap.offer(array[i]);
}
}
}
for (int i = 0; i < k; i++) {
System.out.println(minHeap.poll());
}
}
public static void main(String[] args){
int array[] = {27,15,19,18,28,34,65,49,25,37};
topK(array,3);
}