数据结构之堆排序的java实现

堆是具有以下性质的完全二叉树：

每个结点大于等于它的左右孩子结点的值称为最大堆或大顶堆；

每个结点小于等于它的左右孩子结点的值称为最小堆或小顶堆；

1.具体实现

public class HeapSort {

    public static void main(String[] args){
        int[] a = {3,4,6,1,2,9,8,0,5,7};
        System.out.println(Arrays.toString(sort(a)));
    }

    /**
     * 顺序：  从下往上，从右到左
     * @param arr
     * @return
     */
    private static int[] sort(int[] arr){

        /**
         * 构建最大堆
         */
        for(int i = arr.length/2-1; i >= 0; i--){  //数组从0开始  10个元素  最后一个非叶子结点应该是4
            heapSort(arr, i, arr.length);
        }

        /**
         * 调整堆，每一次都将最大值放置末尾
         */
        for(int i = arr.length-1; i > 0; i--){
            swap(arr, 0, i);
            heapSort(arr, 0, i-1);
        }

        return arr;
    }

    private static void swap(int[] arr, int index1, int index2){    //用于交换根节点与最后一个结点
        int temp = arr[index1];
        arr[index1] = arr[index2];
        arr[index2] = temp;
    }

    

/**
 * 从最后一个非叶子结点开始 很关键
 *  位置图
 *            0
 *        1      2
 *     3   4   5    6
 *  7   8 9  
 * @param arr
 * @param index
 * @param length
 */
    private static void heapSort(int[] arr, int index, int length){

        int temp = arr[index];
        for(int i = index*2 + 1; i < length; i=i*2+1){      //循环用于调整堆，  构建堆可用if
                                                            //+1 很关键  可以画图   
            if(i+1 < length && arr[i] < arr[i+1]){
                ++i;
            }

            if(temp > arr[i]){
                break;
            }

            arr[index] = arr[i];
            index = i;
        }

        arr[index] = temp;
    }
}

2.测试结果

[0, 2, 1, 3, 4, 5, 6, 7, 8, 9]