c++ 堆的vector实现

一、建堆

vector<int> 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<int>());

9 8 2 6 7 0 1 4 3 5 

2.构建最小堆

make_heap(nums.begin(),nums.end(),greater<int>());

 

二、调整堆

包括得到最大值/最小值，插入新元素，删除元素等操作。如果vector使用push_back()插入数据或pop_back()删除数据后，会破坏最大堆/最小堆的性质，所以需要调整堆，常用push_heap()和pop_heap()两个方法。

1、push_heap()用法是，vector先push_back()，后push_heap()：

 

nums.push_back(10);
push_heap(nums.begin(), nums.end(), less<int>());

输出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<int>());
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_back() ？

因为pop_heap()把除了堆顶的元素重新排列成堆，而把堆顶放在vector末尾。