JAVA常见排序算法（学习留痕）-堆排序

将元素调整为大顶堆（降序）或小顶堆（升序），父节点与子节点元素比较，将元素继续调整为大顶堆或小顶堆

public static void main(String[] args) {
        int[] arr = {12,14,16,15,13,18,11,17,10,3,6,7,2,8,9,20};
        heapSort(arr);
        System.out.println("排序完成");
    }

    //执行
    private static void heapSort(int[] arr) {
        //数组调整为大顶堆
        /*for (int i = (arr.length/2) - 1; i >= 0; i--) {
            sortHeap(arr,i,arr.length);
        }
        //交换数据
        for (int i = arr.length-1; i >0 ; i--) {
            swap(arr,0,i);
            sortHeap(arr,0,i);
        }*/
        //数组调整为小顶堆
        for (int i = (arr.length/2) - 1; i >= 0; i--) {
            sortHeap1(arr,i,arr.length);
        }
        //交换数据
        for (int i = arr.length-1; i >0 ; i--) {
            swap(arr,0,i);
            sortHeap1(arr,0,i);
        }
    }

    //值交換
    private static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

    /**
     * 转换为大顶堆
     * @param arr 要转换的数组
     * @param i 节点下标
     * @param length  要转换的区间
     */
    private static void sortHeap(int[] arr, int i, int length) {
        //定义最大父节点
        int temp = arr[i];
        //获取子节点
        for (int j = i*2+1; j < length; j=j*2+1) {
            //右节点存在且值大于左节点
            if (j+1<length && arr[j] < arr[j+1]){
                j++;
            }
            //子节点的值大于最大父节点的值，交换位置，大值放到最大父节点
            if (arr[j] > temp){
                arr[i] = arr[j];
                i = j;
            }else {
                break;
            }
        }
        arr[i] = temp;
    }
    /**
     * 转换为小顶堆
     * @param arr 要转换的数组
     * @param i 节点下标
     * @param length  要转换的区间
     */
    private static void sortHeap1(int[] arr, int i, int length) {
        //定义最小父节点
        int temp = arr[i];
        //获取子节点
        for (int j = i*2+1; j < length; j=j*2+1) {
            //右节点存在且值大于左节点
            if (j+1<length && arr[j] > arr[j+1]){
                j++;
            }
            //子节点的值小于最小父节点的值，交换位置，小值放到最小父节点
            if (arr[j] < temp){
                arr[i] = arr[j];
                i = j;
            }else {
                break;
            }
        }
        arr[i] = temp;
    }