找出数组中最小K个数【最小堆】

面试题 17.14. 最小K个数 - 力扣(LeetCode)

设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例:

复制代码
输入: arr = [1,3,5,7,2,4,6,8], k = 4
输出: [1,2,3,4]

提示:

  • 0 <= len(arr) <= 100000
  • 0 <= k <= min(100000, len(arr))
java 复制代码
public class 最小K个数_堆 {

        public int[] smallestK(int[] arr, int k) {
            // 取前K个数放到数组中
            int[] tree = new int[k];
            for (int i = 0; i < k; i++) {
                tree[i] = arr[i];
            }
            if (k <= 1) {
                return tree;
            }
            // 建立一个最小堆
            buildHeap(tree);
            // 入堆
            for (int i = k; i < arr.length; i++) {
                if (arr[i] <= tree[0]) {
                    tree[0] = arr[i];
                    // 重新排序
                    heapify(tree, 0);
                }
            }

            return tree;
        }

        // 堆排序
        void heapify(int[] tree, int index) {
            // 重新为当前节点及其子七点排序(最小元素正常的过程)
            if (index >= tree.length) {
                return;
            }
            // 记录最小父节点、左子节点及右子节点最小值的下标
            int max = index;
            int left = 2 * index + 1;
            if (left < tree.length && tree[max] < tree[left]) {
                max = left;
            }
            int right = 2 * index + 2;
            if (right < tree.length && tree[max] < tree[right]) {
                max = right;
            }
            // 不是父节点时交换父节点也子节点值后重新为子节点排序
            if (max != index) {
                swap(tree, max, index);
                heapify(tree, max);
            }
        }

        // 构建堆
        void buildHeap(int[] arr) {
            for (int i = (arr.length + 1) / 2; i >= 0; i--) {
                heapify(arr, i);
            }
        }

        // 交换
        void swap(int[] tree, int a, int b) {
            int d = tree[a];
            tree[a] = tree[b];
            tree[b] = d;
        }
    
}
相关推荐
jingfeng5141 分钟前
C++ STL-string类底层实现
前端·c++·算法
雲墨款哥1 小时前
JS算法练习-Day10-判断单调数列
前端·javascript·算法
FPGA1 小时前
CRC校验原理及其FPGA实现
算法
Jina AI1 小时前
回归C++: 在GGUF上构建高效的向量模型
人工智能·算法·机器学习·数据挖掘·回归
Coovally AI模型快速验证1 小时前
YOLO、DarkNet和深度学习如何让自动驾驶看得清?
深度学习·算法·yolo·cnn·自动驾驶·transformer·无人机
luoqice1 小时前
linux下找到指定目录下最新日期log文件
linux·算法
楽码2 小时前
底层技术SwissTable的实现对比
数据结构·后端·算法
fffcccc11122 小时前
初级背包问题,层层剖析为什么这样做。最好需要自己推演一遍。
算法
瓦特what?3 小时前
关于C++的#include的超超超详细讲解
java·开发语言·数据结构·c++·算法·信息可视化·数据挖掘
楽码3 小时前
自动修复GoVet:语言实现对比
后端·算法·编程语言