找出数组中前K小的值&最小堆

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。
代码

public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {

if (input.length==0||k>input.length) {
return new ArrayList<>();
}
ArrayList<Integer> list = new ArrayList<>();
PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
for (int integer : input) {
priorityQueue.add(integer);
}

for (int i = 0; i < k; i++) {
list.add(priorityQueue.poll());
}
return list;
}

    在平时的编程工作中似乎很少碰到PriorityQueue（优先队列） ，故很多人一开始看到优先队列的时候还会有点迷惑。优先队列本质上就是一个最小堆。如果是最大堆，则二叉树的顶点是保存的最大值，最小堆则保存的最小值。最小堆每一个节点都是以该节点为根的子树中的最小值。

    下面是一个典型的优先队列的结构图：



 

    它的每个父节点都比两个子节点要小，但是整个数组又不是完全顺序的。