22 堆排序的思想与代码实现

堆排序

1. 堆排序基本思想

1. 将待排序序列构造成一个大顶堆；
2. 此时，整个序列的最大值就是堆顶的根节点；
3. 将其与末尾元素进行交换，此时末尾就为最大值；
4. 然后将剩余 n-1 个元素重新构造成一个堆，这样会得到 n 个元素的次小值。如此反复执行，便能得到一个有序序列了。

2. 堆排序代码实现

import java.util.Arrays;

public class HeapSort {
    public static void main(String[] args) {
        int[] arr = {4, 6, 8, 5, 9, 2, 10, 1};
        heapSort(arr);
        System.out.println(Arrays.toString(arr));
    }

    /**
     * 堆排序
     * @param arr 待排序的数组
     */
    public static void heapSort(int[] arr) {
        int startIndex = arr.length / 2 - 1;
        // 从倒数第一个非叶子节点开始调整
        for (int i = startIndex; i >= 0; i--) {
            adjustHeap(arr, i, arr.length);
        }
        // 得到大根堆后，交换根节点和最后一个节点的位置，再重新调整，最终得到一个升序的序列
        for (int j = arr.length - 1; j >= 0; j--) {
            int temp = arr[j];
            arr[j] = arr[0];
            arr[0] = temp;
            adjustHeap(arr, 0, j);
        }
    }

    /**
     * 调整为大根堆
     * @param arr 待调整的数组
     * @param i  开始调整的节点
     * @param length 需要调整的数组长度
     */
    private static void adjustHeap(int[] arr, int i, int length) {
        // 临时变量保存当前节点
        int temp = arr[i];
        // 调整当前节点，使得局部成为大根堆
        for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
            // 寻找左右孩子较大的那个
            if (k + 1 < length && arr[k] < arr[k + 1]) {
                k++;
            }
            // 如果当前节点的值小于孩子节点的值，则交换
            if (arr[k] > temp) {
                arr[i] = arr[k];
                i = k;
            } else {
                break;
            }
        }
        arr[i] = temp;
    }
}