Java TopK问题1

题目描述：
从一组数组中选取最大（或最小） K 个元素
思路：

1. 建立一个大小为 k 的小顶堆（如果求最小k对则建大顶堆）
2. 把数放到优先队列里，堆顶为堆中最小数
3. 若取下一个数比较 比堆顶大则入堆 这样结束后堆中即是最大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);
    }