难度：简单

题目：

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

---

示例 1：

输入：arr = [3,2,1], k = 2

输出：[1,2] 或者 [2,1]

---

示例 2：

输入：arr = [0,1,2,1], k = 1

输出：[0]

---

限制：

• 0 <= k <= arr.length <= 10000
• 0 <= arr[i] <= 10000

---

Related Topics

• 数组
• 分治
• 快速选择
• 排序
• 堆（优先队列）

---

重点！！！解题思路

第一步：

只要是找集合最值问题都可以使用堆(优先队列)来解决此题，
由于我们要找的是最小的4个数，所以我们可以采用大顶堆来解决此问题

第二步：

将给定数组中的每个元素加入到堆中，
每次添加的时候都要判断是否大于给定大小，
如果大于给定大小则剔除堆顶元素(堆顶元素为堆中最大值)

第三步：

最后数组访问结束，返回求得的堆

源码：

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(new Comparator<Integer>({
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });
        for (int i:arr){
            priorityQueue.offer(i);
            if (priorityQueue.size()>k){
                priorityQueue.poll();
            }
        }
        return priorityQueue.stream().mapToInt(i->i).toArray();
    }
}

解题结果：

如果您还有什么疑问或解答有问题，可在下方评论，我会及时回复。

系列持续更新中，点个订阅吧