定义
堆:堆数据结构是一种数组对象,它可以被视为一颗完全二叉树结构(或者也有可能是满二叉树)
完全二叉树:若设二叉树的深度为h,除第 h 层外,其它各层 (1~h-1) 的结点数都达到最大个数,第 h 层所有的结点都连续集中在最左边,这就是完全二叉树。
最大堆:任一结点的关键码均大于等于它的左右孩子的关键码,其中堆顶的元素最大。(任一路径中的元素升序排列)
最小堆:任一结点的关键码均小于等于它的左右孩子的关键码,其中堆顶的元素最小。(任一路径中的元素降序排列)
左孩子节点 = 2父节点+1
右孩子节点 = 2父节点+2
父节点 = (孩子节点-1)/ 2
堆的应用:
- 优先级队列
- 堆排序
- 100W个数中找出最大(最小)的前K个数
实现
一、建堆
vector nums = {9, 6, 2, 4, 7, 0, 1, 8, 3, 5};
1、如果使用nums构建最大堆:
make_heap(nums.begin(), nums.end());
//或
make_heap(nums.begin(), nums.end(), less());
输出nums的结果为
9 8 2 6 7 0 1 4 3 5
2、如果使用nums构建最小堆:
make_heap(nums.begin(), nums.end(), greater());
输出nums的结果为
0 3 1 4 5 2 9 8 6 7
输出nums的结果:
//原vector
9 8 2 6 7 0 1 4 3 5
//push_back()后
9 8 2 6 7 0 1 4 3 5 10
//push_heap()后
10 9 2 6 8 0 1 4 3 5 7
2、pop_heap()用法是,先pop_heap(),vector后pop_back():
pop_heap(nums.begin(), nums.end(), less());
nums.pop_back();
输出nums的结果:
//原vector
9 8 2 6 7 0 1 4 3 5
//pop_heap()后
8 7 2 6 5 0 1 4 3 9
//pop_back()后
8 7 2 6 5 0 1 4 3
为什么pop_heap()的用法要反过来呢?
要从我们的目的来考虑,使用pop_heap()的绝大部分目的是要把堆顶元素pop出堆中,因为它最大或最小。如果先用vector的pop_back(),它删除的不是堆顶元素(nums[0]),而是vector的最后一个元素。可见这不是我们想要的结果:对于最大堆,最后一个元素既不是最大,也不一定是最小;对于最小堆,最后一个元素既不是最小,也不一定是最大。pop出来没有意义。
观察pop_heap()对堆做了什么?
pop_heap()把堆顶元素放到了最后一位,然后对它前面的数字重建了堆。这样一来只要再使用pop_back()把最后一位元素删除,就得到了新的堆。
作者:KardelShaw
链接:https://www.jianshu.com/p/13a56502e217
來源:简书
简书著作权归作者所有,任何形式的转载都请联系作者获得授权并注明出处。