找出数组中最小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;
        }
    
}
相关推荐
minji...17 分钟前
C语言 函数递归
c语言·开发语言·算法
你好我是咯咯咯19 分钟前
代码随想录算法训练营Day36
算法
uhakadotcom27 分钟前
如何用AI打造高效招聘系统,HR效率提升100%!
后端·算法·面试
Felven1 小时前
A. Everybody Likes Good Arrays!
数据结构·算法
AI_RSER2 小时前
基于 Google Earth Engine 的南京江宁区土地利用分类(K-Means 聚类)
算法·机器学习·分类·kmeans·聚类·遥感·gee
Small踢倒coffee_氕氘氚3 小时前
是否应该禁止危险运动论文
经验分享·笔记·算法·灌灌灌灌
京东云开发者4 小时前
行稳、致远 | 技术驱动下的思考感悟
算法
Dignity_呱4 小时前
记一次手撕算法面试
前端·算法·面试
CodeJourney.4 小时前
深度探索:DeepSeek赋能WPS图表绘制
数据库·人工智能·算法·信息可视化·excel
陈奕昆4 小时前
6.1腾讯技术岗2025面试趋势前瞻:大模型、云原生与安全隐私新动向
算法·安全·云原生·面试·腾讯