数据结构之堆排序（java）

package com.haiyang.datastructure.sort;

import java.util.Arrays;

/**
 * 堆排序原理：首先将待排序的序列创建为大顶堆，每次取堆顶元素
 * @author haiYang
 * @create 2022-01-18 14:09
 */
public class HeapSort {
    //堆排序
    public static void heapSort(int[] arr){
        int temp=0;
        //将序列初始化为大顶堆
        for (int i = arr.length/2-1; i >=0 ; i--) {
            createBigPileTop(arr,i,arr.length);
        }
        //进行堆排序
        for (int i = arr.length-1; i >0 ; i--) {
        	//将大顶堆根节点元素与最后一个元素交换
            temp = arr[i];
            arr[i] = arr[0];
            arr[0] = temp;
            //将前i个元素创建大顶堆
            createBigPileTop(arr,0,i);
        }
        System.out.println("堆排序结果为：" + Arrays.toString(arr));
    }

    /**
     * 将序列变成大顶堆
     *
     * @param arr    待初始化的数组
     * @param i      最后一个非叶子节点
     * @param length 数组的长度
     */
    public static void createBigPileTop(int[] arr, int i, int length) {
        int temp = arr[i];// 用于保存待调整的元素
        //k为待调整节点的左孩子,k = k * 2 + 1代表要调整的子堆
        for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
            //判断左右孩子的大小，取值大的索引
            if (k + 1 < length && arr[k] < arr[k + 1]) {
                k++;
            }
            //将待调整元素与左右孩子中较大的元素进行比较，用比其大的元素进行替换
            if (temp < arr[k]) {
                arr[i] = arr[k];
                //将子序列继续调整
                i = k;
            } else {
                //如果不满足上诉要求代表当前子序列已经满足大顶堆
                break;
            }
        }
        //将最开始的元素放到满足大顶堆的子序列上
        arr[i] = temp;

    }

    public static void main(String[] args) {
        int[] ints = {4, 6, 8, 5, 9};
        heapSort(ints);
    }
}