堆排序

理解堆排序之前,我们首先需要了解顺序存储二叉树的概念:
假设存在一个数组array,我们从下标0开始,按下标顺序将其按照完全二叉树的形式排列,这样构成的二叉树就称为顺序存储二叉树。顺序存储二叉树的节点array[ i ]的左子节点为array[ 2 * i + 1 ],右子节点为array[ 2 * i + 2 ]。

下面进入正题:
堆其实就是具有特殊性质的完全二叉树:若每个节点的值都大于或等于其左右节点的值,则称为大顶堆;若每个节点的值都小于或等于其左右节点的值,则称为小顶堆。

如果用数组来表示一个大顶堆,那么数组应该满足:
array[ i ] >= array[ 2 * i + 1 ] && array[ i ] >= array[ 2 * i + 2 ] (i从0开始)

如果用数组来表示一个小顶堆,那么数组应该满足:
array[ i ] <= array[ 2 * i + 1 ] && array[ i ] <= array[ 2 * i + 2 ] (i从0开始)

一般升序使用大顶堆,降序使用小顶堆。
在这里插入图片描述
代码实现:

package DataStructure;

import java.util.Arrays;

public class HeapSort {
    public static void main(String[] args) {
        int[] arr = {55, 48, 6, 24, 88, 1230, 558};
        heapSort(arr);
        System.out.println(Arrays.toString(arr));
    }

    /**
     * 将数组调整为大顶堆
     * @param arr 需要调整的数组
     * @param i 从哪个节点开始
     * @param length 需要调整的数组的长度
     */
    public static void heapAdjust(int[] arr, int i, int length) {
        // 存储当前节点的值
        int temp = arr[i];
        // 先查找当前节点的左子节点
        for (int k = 2 * i + 1; k < length; k = 2 * k + 1) {
            // 如果存在右子节点,且比左子节点大,将k替换为右子节点
            // k + 1 < length用来判断右子节点是否存在
            if (k + 1 < length && arr[k + 1] > arr[k]) {
                k++;
            }
            // 若子节点比当前节点大,将当前节点的值替换为子节点的值
            if (arr[k] > temp) {
                arr[i] = arr[k];
                // 此处先不将temp赋给arr[k],因为之后还有可能继续调整,
                // 将k的值赋给i,记录temp需要赋值的位置,
                // 若后续不需要调整,则将temp赋值给arr[i],
                // 若后续发生了调整,则将temp赋值给子节点的子节点(或者更远)
                i = k;
            }else {
                // 如果temp比左右节点都大了,退出循环
                break;
            }
        }
        // 退出循环后,将temp赋值给arr[i]
        arr[i] = temp;
    }

    public static void heapSort(int[] arr) {
        // 从序号最大的非叶子节点开始,将数组调整为大顶堆
        for (int i = arr.length / 2 - 1; i >= 0; i--) {
            heapAdjust(arr, i, arr.length);
        }
        // 进行堆排序,将堆顶元素与数组末尾元素交换,然后将剩下的元素重新调整为大顶堆
        for (int j = arr.length - 1; j > 0; j--) {
            int temp = arr[j];
            arr[j] = arr[0];
            arr[0] = temp;
            // 因为只有堆顶元素发生变换,因此从堆顶即0开始调整,
            // length设置为j可以将交换后的最大值排除在调整范围外。
            heapAdjust(arr, 0, j);
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值