剑指 Offer 40. 最小的k个数（堆排序法）

最新推荐文章于 2025-12-05 14:37:41 发布

原创最新推荐文章于 2025-12-05 14:37:41 发布 · 109 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#堆栈 #算法 #排序算法

排序专栏收录该内容

5 篇文章

订阅专栏

本文介绍了使用Java实现堆排序的详细过程，通过创建一个大根堆并不断调整堆来找出最小的k个数。代码注释清晰，着重讲解了如何利用优先队列（PriorityQueue）进行堆排序，并强调了比较方法的重要性。适用于理解堆排序原理及Java编程实践。

用sort的方法就不写了，没有什么意义
下面代码为堆排序的解法
思路都在代码注解中了，用堆挺简单的，还是调用API，没什么要说的
注意要重优先队列的比较方法

public int[] getLeastNumbers(int[] arr, int k) {
        if (k == 0){
            return new int[0];
        }

        /** 创建一个大根堆，size为k，将数组元素存入堆中，
         * 如果比堆顶还要到，就直接跳过，
         * 如果比堆顶小，就将堆底弹出，将该数放出堆中*/
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>((v1,v2) -> v2 - v1);
        for (int i :
                arr) {
            if (priorityQueue.size() < k)   priorityQueue.add(i);
            else if (priorityQueue.peek() > i){
                priorityQueue.poll();
                priorityQueue.add(i);
            }
        }

        /** 取出对顶元素*/
        int[] result = new int[priorityQueue.size()];
        int i = 0;
        for (int res:
                priorityQueue) {
            result[i++] = res;
        }
        return result;