查看原文
其他

详解堆排序

(点击上方公众号,可快速关注)


转自:静默虚空

http://www.cnblogs.com/jingmoxukong/p/4303826.html

好文投稿, 请点击 → 这里了解详情


堆的概念


堆是一棵顺序存储的完全二叉树。


其中每个结点的关键字都不大于其孩子结点的关键字,这样的堆称为小根堆。


其中每个结点的关键字都不小于其孩子结点的关键字,这样的堆称为大根堆。


举例来说,对于n个元素的序列{R0, R1, ... , Rn}当且仅当满足下列关系之一时,称之为堆:


(1) Ri <= R2i+1 且 Ri <= R2i+2 (小根堆)


(2) Ri >= R2i+1 且 Ri >= R2i+2 (大根堆)


其中i=1,2,…,n/2向下取整; 



如上图所示,序列R{3, 8, 15, 31, 25}是一个典型的小根堆。


堆中有两个父结点,元素3和元素8。


元素3在数组中以R[0]表示,它的左孩子结点是R[1],右孩子结点是R[2]。


元素8在数组中以R[1]表示,它的左孩子结点是R[3],右孩子结点是R[4],它的父结点是R[0]。可以看出,它们满足以下规律:


设当前元素在数组中以R[i]表示,那么,


(1) 它的左孩子结点是:R[2*i+1];


(2) 它的右孩子结点是:R[2*i+2];


(3) 它的父结点是:R[(i-1)/2];


(4) R[i] <= R[2*i+1] 且 R[i] <= R[2i+2]。


要点


首先,按堆的定义将数组R[0..n]调整为堆(这个过程称为创建初始堆),交换R[0]和R[n];然后,将R[0..n-1]调整为堆,交换R[0]和R[n-1];


如此反复,直到交换了R[0]和R[1]为止。 


以上思想可归纳为两个操作:


(1)根据初始数组去构造初始堆(构建一个完全二叉树,保证所有的父结点都比它的孩子结点数值大)。


(2)每次交换第一个和最后一个元素,输出最后一个元素(最大值),然后把剩下元素重新调整为大根堆。 


当输出完最后一个元素后,这个数组已经是按照从小到大的顺序排列了。


先通过详细的实例图来看一下,如何构建初始堆。


设有一个无序序列 { 1, 3, 4, 5, 2, 6, 9, 7, 8, 0 }。



构造了初始堆后,我们来看一下完整的堆排序处理:


还是针对前面提到的无序序列 { 1, 3, 4, 5, 2, 6, 9, 7, 8, 0 } 来加以说明。



相信,通过以上两幅图,应该能很直观的演示堆排序的操作处理。 


核心代码


public void HeapAdjust(int[ ] array, int parent, int length) {

    int temp = array[parent]; // temp保存当前父节点

    int child = 2 * parent + 1; // 先获得左孩子 

    while (child < length) {

        // 如果有右孩子结点,并且右孩子结点的值大于左孩子结点,则选取右孩子结点

        if (child + 1 < length && array[child] < array[child + 1]) {

            child++;

        }

        // 如果父结点的值已经大于孩子结点的值,则直接结束

        if (temp >= array[child])

            break;

        // 把孩子结点的值赋给父结点

        array[parent] = array[child];

        // 选取孩子结点的左孩子结点,继续向下筛选

        parent = child;

        child = 2 * child + 1;

    }

    array[parent] = temp;

}


public void heapSort(int[] list) {

    // 循环建立初始堆

    for (int i = list.length / 2; i >= 0; i--) {

        HeapAdjust(list, i, list.length);

    } 

    // 进行n-1次循环,完成排序

    for (int i = list.length - 1; i > 0; i--) {

        // 最后一个元素和第一元素进行交换

        int temp = list[i];

        list[i] = list[0];

        list[0] = temp;

        // 筛选 R[0] 结点,得到i-1个结点的堆

        HeapAdjust(list, 0, i);

        System.out.format("第 %d 趟: \t", list.length - i);

        printPart(list, 0, list.length - 1);

    }

}


时间复杂度


堆的存储表示是顺序的。因为堆所对应的二叉树为完全二叉树,而完全二叉树通常采用顺序存储方式。


当想得到一个序列中第k个最小的元素之前的部分排序序列,最好采用堆排序。


因为堆排序的时间复杂度是O(n+klog2n),若k≤n/log2n,则可得到的时间复杂度为O(n)。


算法稳定性


堆排序是一种不稳定的排序方法。


因为在堆的调整过程中,关键字进行比较和交换所走的是该结点到叶子结点的一条路径,


因此对于相同的关键字就可能出现排在后面的关键字被交换到前面来的情况。 


完整参考代码


public class HeapSort {

    public void HeapAdjust(int[] array, int parent, int length) {

        int temp = array[parent]; // temp保存当前父节点

        int child = 2 * parent + 1; // 先获得左孩子

        while (child < length) {

            // 如果有右孩子结点,并且右孩子结点的值大于左孩子结点,则选取右孩子结点

            if (child + 1 < length && array[child] < array[child + 1]) {

                child++;

            }

            // 如果父结点的值已经大于孩子结点的值,则直接结束

            if (temp >= array[child])

                break;

            // 把孩子结点的值赋给父结点

            array[parent] = array[child];

            // 选取孩子结点的左孩子结点,继续向下筛选

            parent = child;

            child = 2 * child + 1;

        }

        array[parent] = temp;

    }

    public void heapSort(int[] list) {

        // 循环建立初始堆

        for (int i = list.length / 2; i >= 0; i--) {

            HeapAdjust(list, i, list.length);

        }

        // 进行n-1次循环,完成排序

        for (int i = list.length - 1; i > 0; i--) {

            // 最后一个元素和第一元素进行交换

            int temp = list[i];

            list[i] = list[0];

            list[0] = temp;

            // 筛选 R[0] 结点,得到i-1个结点的堆

            HeapAdjust(list, 0, i);

            System.out.format("第 %d 趟: \t", list.length - i);

            printPart(list, 0, list.length - 1);

        }

    }

    // 打印序列

    public void printPart(int[] list, int begin, int end) {

        for (int i = 0; i < begin; i++) {

            System.out.print("\t");

        }

        for (int i = begin; i <= end; i++) {

            System.out.print(list[i] + "\t");

        }

        System.out.println();

    }

    public static void main(String[] args) {

        // 初始化一个序列

        int[] array = {

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

        };

        // 调用快速排序方法

        HeapSort heap = new HeapSort();

        System.out.print("排序前:\t");

        heap.printPart(array, 0, array.length - 1);

        heap.heapSort(array);

        System.out.print("排序后:\t");

        heap.printPart(array, 0, array.length - 1);

    }

}


执行结果:



参考资料


《数据结构习题与解析》(B级第3版)



觉得本文有帮助?请分享给更多人

关注「算法爱好者」,修炼编程内功

淘口令复制以下红色内容,再打开手淘即可购买

范品社,使用¥极客T恤¥抢先预览(长按复制整段文案,打开手机淘宝即可进入活动内容)

近期,北京地区正常发货,但派件时间有所延长。

    您可能也对以下帖子感兴趣

    文章有问题?点此查看未经处理的缓存