数据结构与算法——堆排序

堆排序介绍

堆的介绍

堆排序利用堆这种数据结构进行排序，是一种选择排序，其时间复杂度是O(n*logn)，时间复杂度很低，它是不稳定排序。

堆：它是具有以下性质的完全二叉树：每个节点的值都>=左右孩子节点的值，称为大顶堆；每个节点的值都<=左右孩子节点的值，称为小顶堆。

大顶堆的一个案例如下图：

小顶堆的一个案例如下图：

在排序时，一般升序采用大顶堆，降序排序采用小顶堆。

堆排序思想

基本思想：（以升序为例）

1) 将待排序序列构造成一个大顶堆——是核心（用顺序存储，数组存放）

2) 此时，整个序列的最大值就是堆顶的根节点。

3) 将其与末尾元素进行交换，此时末尾就为最大值。

4) 然后将剩余 n-1 个元素重新构造成一个堆，这样会得到 n-1 个元素的次小值。如此反复执行，便能得到一个有序序列。

在构建大顶堆的过程中，元素的个数逐渐减少，最后就得到一个有序序列了。

堆排序过程

例如，针对数组arr={4,6,8,5,9}进行堆排序升序排序。

步骤一 构造初始堆。将给定无序序列构造成一个大顶堆：

1）给定无序序列的结构为

2）从最后一个非叶子节点开始（叶子节点不用调整），第一个非叶子节点为arr.length/2-1=1;也就是值为6的节点开始，从左至右，从下到上，进行调整。

可以看到，6的左右节点里，右节点较大，为9，因此将6与9进行交换。

3）找到第2个非叶子节点4，进行调整，4的左右节点里，9较大，且大于4，因此4与9进行交换。

4）再看，4值的节点处，右节点6较大，因此4与6进行交换。

此时，便得到了一个大顶堆。

步骤二 将堆顶元素与末尾元素进行交换，使末尾元素最大。

然后继续调整堆，再将堆顶元素与末尾元素交换，得到第二大元素。如此反复进行交换、重建、交换。

1）将堆顶元素9与末尾元素4进行交换，得到第一大元素9

2）调整结构，使其满足大顶堆

3）再将堆顶元素8与末尾元素5进行交换，得到第二大元素8

4）继续重复上述过程，先调整结构为大顶堆，再交换...直到得到有序序列。

上述就是完整的堆排序的过程。

堆排序代码实现

下面给出堆排序的代码实现

public class HeapSort {
    public static void main(String[] args) {
        //数组升序排列
        int[] arr = {4, 6, 8, 5, 9};
        heapSort(arr);
    }

    //编写排序方法
    public static void heapSort(int[] arr) {
        int temp = 0;
        System.out.println("堆排序！");
        //分步完成
//        adjustHeap(arr, 1, arr.length);
//        System.out.println("第一次调整 " + Arrays.toString(arr));
//
//        adjustHeap(arr, 0, arr.length);
//        System.out.println("第2次调整 " + Arrays.toString(arr));

        //完成最终代码
        //将无序序列构建成一个堆，根据升序降序需求选择大顶堆
        for (int i = arr.length / 2 - 1; i >= 0; i--) {
            adjustHeap(arr, i, arr.length);
        }

        //将堆顶元素与末尾元素交换，将最大元素"沉"到数组末端
        //重新调整结构，使其满足堆定义，然后继续交换堆顶元素与当前末尾元素，反复执行调整+交换步骤，
        //直到整个序列有序。
        for (int j = arr.length -1; j > 0; j--) {
            //交换
            temp = arr[j];
            arr[j] = arr[0];//arr[0]是最大的值
            arr[0] = temp;
            adjustHeap(arr, 0, j);
        }

        System.out.println(Arrays.toString(arr));

    }

    //将一个数组（二叉树），调整成一个大顶堆

    /**
     * 从左到右，从下至上
     * 功能：完成 将 以i对应的非叶子节点 的 树 调整成 大顶堆 （每次都是局部调整）
     * 如对arr[]={4,6,8,5,9} ==> i=1 ==> adjustHeap => 得到{4,9,8,5,6}
     * 再次调用 adjustHeap 传入 i=0 => 从{4,9,8,5,6} adjustHeap => {9,6,8,5,4}
     *
     * @param arr    待调整的数组
     * @param i      表示非叶子节点 在数组中的索引
     * @param length 表示对多少个元素进行调整 ， 因为每次调整的个数逐渐减少
     */
    public static void adjustHeap(int[] arr, int i, int length) {
        int temp = arr[i];//先取出当前元素的值，存在临时变量
        //开始调整
        //k表示 以i为非叶子节点 的左子节点
        for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
            if((k+1) < length && arr[k] < arr[k+1]) {//说明左子节点值<右子节点值
                k++;//让k指向右子节点
            }
            if(arr[k] > temp) {//如果子节点>父节点
                arr[i] = arr[k];//把较大的值赋给当前节点
                i = k;//！！！！！i指向k,继续循环比较
            } else {
                break;
            }
        }
        //for循环结束后，已经将 以i为父节点的树的最大值，放在了最顶上 （局部的）
        arr[i] = temp;//将temp值 放入 调整后的位置  即就是交换
    }
}

代码结果如下：

堆排序的速度相当快！