【八大排序】day2.堆排序

一、堆排序

/**
 * 堆排序：（前提：堆是一颗完全二叉树）
 * 大根堆：每个节点的值都大于等于左右孩子的值
 * 小根堆：每个节点的值都小于等于左右孩子的值
 * 构建大根堆：
 * 比如一个大根堆映射成数组为[50, 45, 40, 20, 25, 35, 30, 10, 15]，
 * 可以发现第一个下标的元素是最大值，将其与末尾元素交换，则末尾元素就是最大值。
 * 堆排序思想：
 * 1.根据初始数组构建堆
 * 2.每次交换第一个和最后一个元素，然后将除最后一个元素外的其他元素重新调整为大根堆
 * 3.重复上述两步，直到没有元素可操作了，就完成排序了
 * 我们需要把一个普通数组转换成大根堆，调整的起点是最后一个非叶子结点，从左到右，
 * 从下到上，继续调整其他非叶子结点，直到根节点为止。
 * 也就是说，要从下向上构建堆，同时要考虑交换以后是否会影响到已经构建好的堆，要进行递归。
 */

public class _6_HeapSort {
    public static void main(String[] args) {
        Integer[] arr = {3, 5, 1, 7, 2, 3, 4, 8};
        heapSort(arr);
        System.out.println(Arrays.toString(arr));
    }

    /**
     * 堆排序
     */
    public static void heapSort(Integer[] arr) {
        //找到最后一个非叶子结点，也就是最后一个结点的父节点
        int start = (arr.length - 1) / 2;
        //从下到上，调整为大根堆
        for (int i = start; i >= 0; i--) {
            maxHeap(arr, arr.length, i);
        }
        //先把数组中第0个位置的数和堆中最后一个位置的数交换位置，再把前面的处理成大根堆
        for (int i = arr.length - 1; i > 0; i--) {
            int temp = arr[0];
            arr[0] = arr[i];
            arr[i] = temp;
            maxHeap(arr, i, 0);
        }

    }

    /**
     * 转换为大根堆
     */
    public static void maxHeap(Integer[] arr, int size, int index) {
        //左子节点
        int left = 2 * index + 1;
        //右子节点
        int right = 2 * index + 2;
        //记录最大值的下标
        int max = index;
        //max和两个子节点分别比较，找出最大的节点下标
        if (left < size && arr[left] > arr[max]) {
            max = left;
        }
        if (right < size && arr[right] > arr[max]) {
            max = right;
        }
        //交换位置
        if (max != index) {
            int temp = arr[max];
            arr[max] = arr[index];
            arr[index] = temp;
            //因为交换位置后可能导致子树不满足大根堆条件，所以要递归子树
            maxHeap(arr, size, max);
        }
    }
}

注意事项：

1.堆排序分为构建堆和堆排序两个部分

2.构建堆时，不管是大根堆还是小根堆，都需要自底向上构建，从最后一个非叶子结点开始，也就是最后一个节点的父节点，(arr.length-1)/2就是他的数组下标

3.构建堆时，因为是自下而上，当发生结点之间的交换时，可能会导致连锁反应，所以在交换完毕后，需要对交换的子节点位置进行递归，继续构建堆。

4.堆排序时，把arr[0]与arr[end]交换，这样最大值就去了最后，然后再构建0~end-1的大根堆，再取出根节点与最后一个的上一个位置互换，直到最后剩下一个元素，无法进行互换操作时，排序完成。