方便建堆函数C++

对make_heap(), pop_heap(), push_heap()的用法做个总结:
make_heap()生成堆,他有两个参数,也可以有三个参数,前两个参数是指向开始元素的迭代器和指向结束元素的下一个元素的迭代器。第三个参数是可选的,可以用伪函数less()和greater()来生成大顶堆和小顶堆,其中type为元素类型。如果只传入前两个参数,默认是生成大顶堆。
push_heap()是在堆的基础上进行数据的插入操作,参数与make_heap()相同,需要注意的是,只有make_heap()和push_heap()同为大顶堆或小顶堆,才能插入。
pop_heap()是在堆的基础上,弹出堆顶元素。这里需要注意的是,pop_heap()并没有删除元素,而是将堆顶元素和数组最后一个元素进行了替换,如果要删除这个元素,还需要对数组进行pop_back()操作。
若用make_heap(), pop_heap(), push_heap(),需要添加头文件 # include 。
用less ()和greater () 需要添加头文件 # include 。

这里只是描述用法,粘贴代码如下:

# include 
# include 
# include 
# include 

using namespace std;

void printVec(vector 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};
    vector nums(nums_temp, nums_temp + 8);
    cout << "sorce nums: ";
    printVec(nums);

    cout << "(默认)make_heap: ";
    make_heap(nums.begin(), nums.end());
    printVec(nums);

    cout << "(less)make_heap: ";
    make_heap(nums.begin(), nums.end(), less ());
    printVec(nums);

    cout << "(greater)make_heap: ";
    make_heap(nums.begin(), nums.end(), greater ());
    printVec(nums);

    cout << "此时,nums为小顶堆" << endl;;
    cout << "push_back(3)" << endl;
    nums.push_back(3);
    cout << "忽略第三个参数,即为默认)push_heap: ";
    push_heap(nums.begin(), nums.end());
    printVec(nums);
    cout << "第三个参数为greater: ";
    push_heap(nums.begin(), nums.end(), greater());
    printVec(nums);
    cout << "(做替换)pop_heap: ";
    pop_heap(nums.begin(), nums.end());
    printVec(nums);
    cout << "pop_back(): ";
    nums.pop_back();
    printVec(nums);
}
结果为:
堆的运行结果
这个是为了记录如何用堆,若有不对之处,敬请指正。。

你可能感兴趣的:(堆和二叉树)