图解堆排序

一、引言二、图解堆排序(heapsort)三、java代码实现及时间复杂度分析四、总结

一、引言

优先队列可以用于以O(NlogN)时间排序,正如上一篇的求解topK问题中用到的思想一样,这种思想就是堆排序(heapsort)。

二、图解堆排序(heapsort)

  1. 算法思想:通过将数组元素进行buildHeap进行堆序化(构建大顶堆);再对堆进行N-1次deleteMax操作。这里有个技巧,如果使用新的数组来存储,需要O(N)的空间;但每次deleteMax会空出一个位置,并将尾端的节点进行下滤操作,那我们就可以将deleteMax的数据放到尾端。
  2. 堆排序过程:

对二叉堆不了解的可以看看图解优先队列(堆)

三、java代码实现及时间复杂度分析

我们从数组下标0开始,不像二叉堆从数组下标1开始。

  • 代码实现
public class Heapsort {
    public static void main(String[] args) {
        Integer[] integers = {711391158};
        System.out.println("原序列:" + Arrays.toString(integers));
        heapsort(integers);
        System.out.println("排序后:" + Arrays.toString(integers));
    }

    public static <T extends Comparable<? super T>> void heapsort(T[] a) {
        if (null == a || a.length == 0) {
            throw new RuntimeException("数组为null或长度为0");
        }
        //构建堆
        for (int i = a.length / 2 - 1; i >= 0; i--) {
            percDown(a, i, a.length);
        }
        //deleteMax
        for (int i = a.length - 1; i > 0; i--) {
            swapReferences(a, 0, i);
            percDown(a, 0, i);
        }
    }

    /**
     * 下滤的方法
     *
     * @param a:待排序数组
     * @param i:从哪个索引开始下滤
     * @param n           :二叉堆的逻辑大小
     * @param <T>
     */

    private static <T extends Comparable<? super T>> void percDown(T[] a, int i, int n) {
        int child;
        T tmp;
        for (tmp = a[i]; leftChild(i) < n; i = child) {
            child = leftChild(i);
            if (child != n - 1 && a[child].compareTo(a[child + 1]) < 0) {
                child++;
            }
            if (tmp.compareTo(a[child]) < 0) {
                a[i] = a[child];
            } else {
                break;
            }
        }
        a[i] = tmp;
    }

    private static int leftChild(int i) {
        return 2 * i + 1;
    }

    /**
     * 交换数组中两个位置的元素
     *
     * @param a:目标数组
     * @param index1 :第一个元素下标
     * @param index2 :第二个元素下标
     * @param <T>
     */

    private static <T> void swapReferences(T[] a, int index1, int index2) {
        T tmp = a[index1];
        a[index1] = a[index2];
        a[index2] = tmp;
    }
}
//输出结果
//原序列:[7, 1, 13, 9, 11, 5, 8]
//排序后:[1, 5, 7, 8, 9, 11, 13]
  • 时间复杂度:buildHeap使用O(N)的时间,元素下滤需要O(logN),需要下滤N-1次,所以总共需要O(N+(N-1)logN) = O(NlogN)。从过程可以看出,堆排序,不管最好,最坏时间复杂度都稳定在O(NlogN)。

  • 空间复杂度:使用自身存储,无疑是O(1)

四、总结

本篇通过画图,说明堆排序的过程,清晰明了知道堆排序先经过堆序化,再通过deleteMax进行排序。其空间复杂度是O(1),时间复杂度稳定在O(NlogN)。

原文地址:https://www.cnblogs.com/9dragon/p/10778074.html