最小的k个数

SeaSky_Steven

于 2021-03-07 18:04:39 发布

阅读量245

点赞数

分类专栏：数据结构与算法文章标签：堆排序大根堆最小的k个数排序数组

本文链接：https://blog.csdn.net/SeaSky_Steven/article/details/114491508

版权

数据结构与算法专栏收录该内容

71 篇文章 1 订阅

订阅专栏

该博客介绍了如何利用Java的PriorityQueue实现从整数数组中找出最小的k个数。通过建立大根堆，先将前k个数放入堆中，然后遍历剩余元素，如果新元素小于堆顶元素，则替换并调整堆。这种方法的时间复杂度为O(nlogk)，空间复杂度为O(k)。示例展示了在不同输入下如何得到最小的k个数。

摘要由CSDN通过智能技术生成

DESC:

输入整数数组 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

来源：力扣（LeetCode）

CODE:

JAVA:

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if (arr == null || k>arr.length || k==0) {
            return new int[0];
        }

        PriorityQueue<Integer> queue = new PriorityQueue(new Comparator<Integer>(){
            public int compare(Integer v1, Integer v2) {
                return v2.compareTo(v1);
            }
        });

        for (int i=0; i<k;i++) {
            queue.offer(arr[i]);
        }

        for (int i=k; i<arr.length; i++) {
            if (queue.peek() > arr[i]) {
                queue.poll();
                queue.offer(arr[i]);
            }
        }
        int[] res = new int[k];
        for (int i=0; i<k; i++) {
            res[i] = queue.poll();
        }
        return res;
    }
}