HeapSort堆排序（罗召勇版）

package demo030102;

import java.util.Arrays;

public class HeapSort {
    public static void main(String[] args) {
        int[] arr = new int[]{22, 5, 10, 11, 6, 23, 54, 3, 8};
        heapSort(arr);

        System.out.println(Arrays.toString(arr));
    }

    public static void heapSort(int[] arr) {
        //开始位置为最后一个节点的父节点
        //第一次循环操作，调整数组的二叉树结构为大顶堆
        for (int start = (arr.length - 1) / 2; start >= 0; start--) {
            maxHeap(arr, arr.length, start);
        }
        //把数组中第一个元素和最后一个交换，递归调用生成新的大顶堆
        for (int i = arr.length - 1; i > 0; i--) {
            int temp = arr[0];
            arr[0] = arr[i];
            arr[i] = temp;
            //递归，在大顶堆框架的基础上，只处理root节点即可！！！
            maxHeap(arr, i, 0);
        }
    }

    /**
     * @param arr      传入一个数组
     * @param heapSize 待排序数组大小
     * @param index    待排序节点根节点位置，最后一个非叶子节点，最后一个节点的父节点
     */
    //堆排序，每次从数组中取出最大值放在数组最后的位置
    public static void maxHeap(int[] arr, int heapSize, int index) {
        //左子节点
        int leftNode = 2 * index + 1;
        //右子节点
        int rightNode = 2 * index + 2;
        //root和左右节点分别对比，找出最大节点data,设置浮标temp
        int max = index;
        if (leftNode < heapSize && arr[max] < arr[leftNode]) {
            max = leftNode;
        }
        if (rightNode < heapSize && arr[max] < arr[rightNode]) {
            max = rightNode;
        }
        if (max != index) {
            int val = arr[max];
            arr[max] = arr[index];
            arr[index] = val;
            //交换位置后，可能会破坏底层排好的堆，需要递归调用自身修正
            maxHeap(arr, heapSize, max);
        }
    }
}