java---堆排序---数组模拟实现（每日一道算法2022.8.14）

最新推荐文章于 2024-10-06 14:10:45 发布

SRestia

最新推荐文章于 2024-10-06 14:10:45 发布

阅读量259

点赞数 1

分类专栏：算法文章标签：算法 java 开发语言

本文链接：https://blog.csdn.net/SRestia/article/details/126333888

版权

算法专栏收录该内容

157 篇文章 5 订阅

订阅专栏

题目
输入一个长度为 n 的整数数列，从小到大输出前 m 小的数

输入格式
第一行包含整数 n 和 m
第二行包含 n 个整数，表示整数数列

输入：
分别是数组的元素数量和求最小值的次数
5 3
数组
4 5 1 3 2

输出：
1 2 3

public class 堆_堆排序{
    //初始化,切记这里size需要先初始化，要不然不是全局变量在down里面用不到
    public static int N = 100010, size;
    public static int[] heap = new int[N];

    public static void main(String[] args) {
        //初始化，这里i从1开始，因为堆的左右节点是当前节点*2和*2+1来计算的，0开始的话不太方便
        Scanner in = new Scanner(System.in);
        int n = in.nextInt(), m = in.nextInt();
        size = n;
        
        //导入数据到数组
        for (int i = 1; i<=n; i++) {heap[i] = in.nextInt();}
        //使用down将二叉树进行排序，变为堆
        for (int i = n/2; i>0; i--) {down(i);}
        
        //执行操作，每次拿到最小的，然后删除掉，再down将整个二叉树排序为堆
        while (m-- > 0) {
            System.out.print(heap[1] + " ");
            heap[1] = heap[size--];
            down(1);
        }
    }
    public static void down(int u) {
        int t = u;
        //拿到三个数中最小的
        if (u*2<=size && heap[u*2]<heap[t]) {t = u*2;}
        if (u*2+1<=size && heap[u*2+1]<heap[t]) {t = u*2+1;}
        //如果最小的数不是根，那么就和根交换
        if (t!=u) {
            int temp = heap[u];
            heap[u] = heap[t];
            heap[t] = temp;
            //递归将改动影响到的部分继续down处理
            down(t);
        }
    }
}