码迷,mamicode.com
首页 > 其他好文 > 详细

随便说说堆(一)——二叉堆

时间:2018-12-22 17:28:49      阅读:187      评论:0      收藏:0      [点我收藏+]

标签:例子   image   for   int   swa   技术   等于   新建   ems   

二叉堆可以被看作是一个数组,也可以简单的看作是一个近似的完全二叉树,二叉堆有最大堆和最小堆,分别具有堆的性质:最大堆的某个结点的值最多与其父结点一样大,最小堆则是某个结点的值最多与其父结点一样小。所以最大堆中最大的结点永远是根结点,最小堆中最小的结点永远是根节点。

既然二叉堆是一种数据结构,就有其支持的操作(这里以最小堆为例):

  • make_heap:建立一个空堆,或者把数组中元素转换成二叉堆。
  • insert:插入元素。
  • minimun:返回一个最小数。
  • extract_min:移除最小结点。
  • union:合并堆

make_heap

先给一组数{27,17,3,16,13,10,1,5,7}输入数组,数组下标从0开始。然后我们画出这棵树:

技术分享图片

然后根据近似满二叉树的性质,有n个结点,[n/2]+1,[n/2]+2,……,n都是叶结点。然后可以通过对除了叶结点以外的结点i(0到n/2)进行一次下滤的操作,若儿子结点有小于结点i的结点,将儿子结点中最小者与结点i交换,再与交换后的位置的儿子结点继续比较,直到小于儿子结点或者为叶结点为止。遍历完,就得到最小堆。 时间复杂度建立空堆O(1),立地建堆O(n)。

图示

技术分享图片

代码实现

 1 //维护堆
 2 void max_heapify(int *a, int i) {
 3     int l = 2 * i + 1;
 4     int r = 2 * i + 2;
 5     int largest;
 6     if (a[l] != -1 && a[l] < a[i]) largest = l;
 7     else largest = i;
 8     if (a[r] != -1 && a[r] < a[largest]) largest = r;
 9     if (largest != i) {
10         swap(a[i], a[largest]);
11         max_heapify(a, largest);
12     }
13 }
14 //建堆
15 void make_heap(int *a, int n) {
16     for (int i = n / 2; i >= 0; i--) max_heapify(a, i);
17 }

 

 


insert

对于插入操作,首先将元素push到数组尾部,然后将其进行上滤操作,也就是将其与父结点比较,如果小于父结点就交换,直到大于等于父结点或者到达根。时间复杂度为O(logn)

举个例子:在{1,5,3,7,13,10,27,16,17}中插入4。

图示

技术分享图片

代码实现

1 //n是数组长度
2 void insert(int *a, int num, int &n) {
3     a[n++] = num;
4     for (int i = n - 1; i >= 0; i = (i - 1) / 2) {
5         if (a[i] < a[(i - 1) / 2]) swap(a[i], a[(i - 1) / 2]);
6         else break;
7     }
8 }

 

 


minimun

返回最小数,对于最小堆来说返回根即可。时间复杂度O(1)

代码实现

1 int minimun(int *a) {
2     return a[0];
3 }

 


extract_min

移除最小顶点,也就是最小堆的根,此时需要将堆最后一个叶节点摘下替换掉根,这样不会破坏近似满二叉树的结构,然后从上至下更新堆,维护堆的性质。时间复杂度O(logn)

图示

技术分享图片

代码实现

1 void extract_min(int *a, int &n) {
2     a[0] = a[n - 1];
3     n--;
4     max_heapify(a, 0);
5 }

 


union

合并堆的话其实就相当于把两个堆数组合并,然后重新建堆。所以时间复杂度也是线性的,为O(n)。

代码实现

1 void Union(int *a, int *b, int &n, int &m) {
2     for (int i = 0; i < m; i++) {
3         a[n++] = b[i];
4     }
5     m = 0;
6     memset(b, -1, sizeof(b));
7     for (int i = n / 2; i >= 0; i--) max_heapify(a, i);
8 }

 

这里就说完了二叉堆,下一篇在随便说说堆中的二项堆。

随便说说堆(一)——二叉堆

标签:例子   image   for   int   swa   技术   等于   新建   ems   

原文地址:https://www.cnblogs.com/pullself/p/10161645.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!