堆排序Java实现

二叉树

每个节点至多有两个子节点的树形结构。

完全二叉树

从上到下，从左至右排布的二叉树（最后一个节点之前没有空节点）。完全二叉树可以使用数组来表示。

二叉堆

一棵父子节点之间存在大小关系的完全二叉树。堆排序使用到的就是二叉堆。

如何使用二叉堆排序

将给定的数组建堆（升序大根堆，降序小根堆），交换数组的第0个元素和最后一个元素，对数组去除掉最后一个元素的部分做为新数组建堆，重复上述步骤，知道数组完全有序。

 堆排序使用Java代码实现

/*
当前节点下标为i时
父节点的下标为(i - 1) / 2
左子节点的下标为 2 * i + 1
右子节点的下标为 2 * (i + 1)
最后一个非叶子节点的下标为 end / 2 - 1
 */
public class HeapSort {

    /**
     * 完成大根堆一个分支的交换
     * @param heap 需要建堆的数组
     * @param start 需要建堆的起始索引
     * @param end 需要建堆的结束索引 + 1
     */
    public static void shift(int[] heap,int start,int end){
        //暂存起始值
        int temp = heap[start];
        //假设需要交换的值为左子节点
        int swapIndex = start * 2 + 1;

        //左子节点的位置如果比最后一个位置大，则右子节点的位置也超出
        while (swapIndex < end){
            //右子节点存在，且右子节点大于左子节点，替换需要交换的节点为右子节点
            if (swapIndex + 1 < end && heap[swapIndex + 1] > heap[swapIndex]){
                swapIndex++;
            }
            if (heap[swapIndex] <= heap[start]){
                break;
            }
            heap[start] = heap[swapIndex];
            //交换节点做为下一个起始节点
            start = swapIndex;
            swapIndex = start * 2 + 1;
            heap[start] = temp;
        }

    }

    /**
     * 建堆
     * @param heap 数组
     * @param length 建堆长度
     */
    public static void createHeap(int[] heap,int length){
        for (int i = length / 2 - 1; i >= 0; i--){
            shift(heap,i,length);
        }
    }

    /**
     * 实现堆排序
     * @param array 需要排序的数组
     */
    public static void sort(int[] array){
        //最后的下标
        int last = array.length  - 1;
        //建堆
        createHeap(array, array.length);

        for (int i = array.length  - 1; i > 0; i--){
            //将堆顶元素放到数组末尾
            int temp = array[0];
            array[0] = array[last];
            array[last] = temp;
            //缩短长度
            last--;
            //对剩余数组建堆
            shift(array,0,last + 1);
        }
    }



    public static void main(String[] args) {
        int[] array = {66,52,3,86,93,2,1};
        sort(array);
        System.out.println(Arrays.toString(array));
    }

}

运行结果