找出数组中最小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;
        }
    
}
相关推荐
晓纪同学15 分钟前
随性研究c++-智能指针
开发语言·c++·算法
程序员爱钓鱼30 分钟前
Go 连接 Oracle 太麻烦?一文教你优雅搞定 GORM + Oracle 全流程!
后端·算法·go
xuanjiong41 分钟前
纯个人整理,蓝桥杯使用的算法模板day4(图论 最小生成树问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·图论
lmy201211082 小时前
提高:图论:强连通分量 图的遍历
c++·算法·图论·强联通分量
人类群星闪耀时2 小时前
破解 N 皇后 II:位运算的高效艺术
python·算法·数学建模
Demons_kirit2 小时前
LeetCode 1863.找出所有子集的异或总和再求和
数据结构·算法·leetcode
竹下为生3 小时前
LeetCode --- 443周赛
算法·leetcode·职场和发展
green5+13 小时前
LeetCode18四数之和
java·开发语言·算法
雾里看山3 小时前
算法思想之双指针(一)
算法·leetcode·推荐算法
2401_827499993 小时前
leetcode-热题100(3)
数据结构·算法·leetcode