《算法导论》Chapter6 堆

来源:互联网 发布:mac怎么批量删除文件 编辑:程序博客网 时间:2024/06/05 03:31

先介绍 堆的几个基本性质:

MAX-HEAPIFY过程,时间复杂度为O(lgn)。他是维护最大堆性质的关键;

BUILD-MAX-HEAP过程,具有线性时间复杂度,功能是从无序的输入数据数组中构造一个最大堆;

HEAPSORT过程,时间复杂度为O(nlgn),其功能是对一个数组进行原址排序;

MAX-HEAP-INSERT,HEAP-EXTRACT-MAX,HEAP-INCREASE-KEY,HEAP-MAXIMUM过程时间复杂度为O(lgn),功能是利用堆实现一个优先队列。

6.1-1 高度为h的堆中,元素的个数范围为:2.^(h-1) <= n(h) <=2.^h-1。


建堆:对于一个n个数据的数组A,其中A(n/2+1, ..., n)(如n=15,则从n=8-15)中的元素都是树的叶子;建堆过程就是对A(n/2+1, ..., n)中的每个元素(叶节点),逆序调用一次MAX-HEAPIFY(最大值从下往上不断跑到根节点),建的堆是个最大堆。时间复杂度为O(n);


堆排序算法

特点:堆排序是一种选择排序算法,与关键字的初始排列次序无关,即就是在最好,最坏,一般的情况下排序时间复杂度不变。对包含n个数的输入数组,平均时间为O(nlgn),最坏情况(已经排好序)也是是O(nlgn),最好情况(完全无序)也是O(nlgn)。

下面是堆排序算法的代码实现:

        // 堆排序算法,结合下面两个算法的时间复杂度分析,可以看出堆排序算法时间复杂度为O(nlogn)public void HeapSort(int[] a) {BuildMaxHeap(a);for (int i = a.length - 1; i >= 1; i--) {swap(a, 0, i);MaxHeapify(a, 0, i);}}// 交换数组中的元素public void swap(int[] a, int i, int j) {int temp = a[i];a[i] = a[j];a[j] = temp;}// 建(最大)堆,时间复杂度为O(nlogn),该方法调用了n次MaxHeapify;public void BuildMaxHeap(int[] a) {for (int i = a.length / 2 - 1; i >= 0; i--) {MaxHeapify(a, i, a.length);}}// 维护最大堆性质,该方法时间复杂度为O(logn),// 设堆的高度为h,则时间复杂度为O(h);// 此处,增加一个参数max,是为了适应在HeapSort方法中的对大小的变化(逐渐减小)。public void MaxHeapify(int[] a, int i, int max) {int l = 2 * i + 1; // 节点i左侧子节点int r = l + 1; // 节点i右侧子节点int largest = l; // 最大值下标if (l < max && a[l] > a[i]) {largest = l;} else {largest = i;}if (r < max && a[r] > a[largest]) {largest = r;}if (largest != i) {swap(a, largest, i);MaxHeapify(a, largest, max);}}


逻辑就是:先把数组构建成最大堆,由于最大堆的性质,数组第一个元素肯定是最大的,就把最大的元素a[1]和a[n]互换。之后把数组长度减1,对n-1个数组元素重新检查最大堆性质,把最大的元素重新放大a[1]位置,再次把a[1]和a[n-1]互换,这样依次循环,就把所有元素按照从小到大的顺序排列了。如果需要从大到小,可以构建最小堆,过程类似。上述代码每一个重点都有相应的注释,可以轻易看懂。


堆的一个应用:优先队列(最大优先队列和最小优先队列)

优先队列是一种用来维护由一组元素构成的集合S的数据结构,其中的每一原色都有一个相关的值,称为关键字(key)。一个最大优先队列支持以下操作:

INSERT(S,x):把元素插入到集合S中;

MAXIMUM(S):返回S中具有最大关键字的元素;

EXTRACT-MAX(S):去掉并返回S中的具有最大关键字的元素;

INCREASE-KEY(S,x,k):将元素x的关键字值增加到k;

最大优先队列有很多应用,其中一个就是在共享计算机系统中的作业调度

最小优先队列可被应用与基于事件驱动的模拟器;

        //MAXIMUM函数,返回堆最大值public int MaxiMum(int[] a){return a[0];}//抽取出队中最大的键值,之后继续维护为最大对序列public int ExtractMax(int[] a){int max = a[0];    a[0] = a[a.length - 1];    int[] newA = new int[a.length - 1];    System.arraycopy(a, 0, newA, 0, newA.length);    MaxHeapify(newA, 0, newA.length);    return max;}

heapIncreaseKey(i, key)会增大下标为i的元素为key。首先将a[i]的值更新为key,因为增大的a[i]关键字可能会违背最大堆的性质,因此我们需要对a[i]进行逐级上升。即将当前元素逐级与父节点比较,如果大于父节点,则与父节点进行交换,一直到当前元素小于父节点为止:

public void heapIncreaseKey(int i, int key) {    if (key > a[i]) {        a[i] = key;        while (i > 0 && a[parent(i)] < a[i]) {  //逐级上升            swap(a, parent(i), i);            i = parent(i);        }    } else {        throw new IllegalArgumentException("key is too small");    }}private int parent(int i) {    return (i + 1) / 2 - 1;}

maxHeapInsert(key)十分简单,因为它等价于数组长度增加一,然后最后一个元素设置为-∞,然后把它增大为key的操作:

public void maxHeapInsert(int key) {    int[] newA = new int[a.length + 1];    System.arraycopy(a, 0, newA, 0, a.length);    newA[newA.length - 1] = Integer.MIN_VALUE;    this.a = newA;    heapIncreaseKey(a.length - 1, key);}

在一个包含n个元素的堆中,所有优先队列的操作都可以在O(logn)时间内完成。


本文后半部分参考了该博文:

https://lufficc.com/blog/heap-sort-and-max-priority-queue











原创粉丝点击