写在前面
最近渐渐爱上写博客,觉得每天学到的知识需要保鲜,写的源码也能及时与大家分享,接下来进入正题。
最大堆(最小堆)
最大堆(最小堆)是非常重要的数据结构,公司面试时经常会被问到,在这里,我不会详细介绍它的原理,而是介绍它的适用场景以及两种写法,对原理不了解的可以查看:
https://en.wikipedia.org/wiki/Heap_%28data_structure%29
假设有10万个数,让你求最小的k个数,你会怎么做?
- 先排序,然后取前k个数,时间复杂度O(nlogn)
- 维护k个数的最大堆,时间复杂度O(nlogk)
考虑第一种方式,当内存有限时,没法一次载入10万个数,排序不可取,而第二种方式,每次只载入一个数,最多存储k个数(最大堆),当堆满时,通过比较当前数与堆顶值,若堆顶值大,则删除堆顶,插入当前数,反之,则不做任何操作,也就是说,最大堆一直存储着当前载入数里最小的k个数。
最大堆C++实现源码
这里给出了两种写法,最小堆同理,值得注意的是,最大堆(最小堆)一定是一颗完全二叉树,可以用数组表示,如有错误还望指正。
- 非递归写法
#include <vector>
#include <cassert>
using namespace std;
class MaxHeap
{
private:
vector<int> heap;
int size;
public:
void make_heap(vecto