make_heap() pop_heap() push_heap()用法
文章转载
1.make_heap()是生成一个堆,大顶堆或小顶堆
- make_heap(_RAIter,_RAIter) 默认生成大顶堆
- make_heap(_RAIter,_RAIter,_Compare) _Compare有两种参数,一种是greater(生成小顶堆),一种是less(生成大顶堆)
2.push_heap()是向堆中插入一个元素,并且使堆的规则依然成立
- push_heap(_RAIter,_RAIter) 默认为大顶堆
- push_heap(_RAIter,_RAIter,_Compare) _Compare有两种参数,一种是greater(小顶堆),一种是less(大顶堆)
- 调用push_heap之前必须调用make_heap创建一个堆
- 首先数组push_back插入元素,然后再调用push_heap,它会使最后一个元素插到合适位置
- 注意,push_heap中的_Compare和make_heap中的_Compare参数必须是一致的,不然会插入堆失败,最后一个元素还是在最后位置,导致插入失败
3.pop_heap()是在堆的基础上,弹出堆顶元素。
- pop_heap(_RAIter,_RAIter) 默认为大顶堆
- pop_heap(_RAIter,_RAIter,_Compare) _Compare有两种参数,一种是greater(小顶堆),一种是less(大顶堆)
- 比如pop_heap(nums.begin(), nums.end(),greater
()),它会将堆顶元素(即为数组第一个位置)和数组最后一个位置对调,然后你可以调用数组pop_back,删除这个元素 - 注意,pop_heap中的_Compare和make_heap中的_Compare参数必须是一致的,不然会失败
代码
# include <iostream>
# include <functional>
# include <vector>
# include <algorithm>
using namespace std;
void printVec(vector<int> nums)
{
for (int i = 0; i < nums.size(); ++i)
cout << nums[i] << " ";
cout << endl;
}
int main(void)
{
int nums_temp[] = {8, 3, 4, 8, 9, 2, 3, 4, 10};
vector<int> nums(nums_temp, nums_temp + 9);
cout << "make_heap之前: ";
printVec(nums);
cout << "(默认(less))make_heap: ";
make_heap(nums.begin(), nums.end());
printVec(nums);
cout << "(less)make_heap: ";
make_heap(nums.begin(), nums.end(), less<int> ());
printVec(nums);
cout << "(greater)make_heap: ";
make_heap(nums.begin(), nums.end(), greater<int> ());
printVec(nums);
cout << "此时,nums为小顶堆 greater" << endl;
cout << "push_back(3)" << endl;
nums.push_back(3);
cout << "默认(less)push_heap 此时push_heap失败: ";
push_heap(nums.begin(), nums.end());
printVec(nums);
cout << "push_heap为greater 和make_heap一致,此时push_heap成功: ";
push_heap(nums.begin(), nums.end(), greater<int>());
printVec(nums);
cout << "(greater,不然会失败)pop_heap: ";
pop_heap(nums.begin(), nums.end(),greater<int>());
printVec(nums);
cout << "pop_back(): ";
nums.pop_back();
printVec(nums);
}
结果
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 时间海!
评论