找出数组中最小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;
        }
    
}
相关推荐
yyt_cdeyyds5 分钟前
FIFO和LRU算法实现操作系统中主存管理
算法
alphaTao32 分钟前
LeetCode 每日一题 2024/11/18-2024/11/24
算法·leetcode
kitesxian41 分钟前
Leetcode448. 找到所有数组中消失的数字(HOT100)+Leetcode139. 单词拆分(HOT100)
数据结构·算法·leetcode
VertexGeek1 小时前
Rust学习(八):异常处理和宏编程:
学习·算法·rust
石小石Orz1 小时前
Three.js + AI:AI 算法生成 3D 萤火虫飞舞效果~
javascript·人工智能·算法
jiao_mrswang2 小时前
leetcode-18-四数之和
算法·leetcode·职场和发展
qystca2 小时前
洛谷 B3637 最长上升子序列 C语言 记忆化搜索->‘正序‘dp
c语言·开发语言·算法
薯条不要番茄酱2 小时前
数据结构-8.Java. 七大排序算法(中篇)
java·开发语言·数据结构·后端·算法·排序算法·intellij-idea
今天吃饺子2 小时前
2024年SCI一区最新改进优化算法——四参数自适应生长优化器,MATLAB代码免费获取...
开发语言·算法·matlab
是阿建吖!2 小时前
【优选算法】二分查找
c++·算法