合并果子（STL 堆相关操作）

最新推荐文章于 2024-07-27 19:37:05 发布

Sc Turing

最新推荐文章于 2024-07-27 19:37:05 发布

阅读量124

点赞数

分类专栏： PAT 甲级文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/m0_62960816/article/details/127466397

版权

PAT 甲级专栏收录该内容

12 篇文章 0 订阅

订阅专栏

本文介绍了如何使用C++的vector和make_heap/push_heap/pop_heap函数来创建、维护和操作小顶堆。通过实例展示了如何插入元素并保持堆的性质，以及如何弹出堆顶元素进行后续处理。

摘要由CSDN通过智能技术生成

vector与堆

函数说明：

make_heap()是生成一个堆，大顶堆或小顶堆
make_heap(_RAIter,_RAIter) 默认生成大顶堆
make_heap(_RAIter,_RAIter,_Compare) _Compare有两种参数，一种是greater（生成小顶堆），一种是less（生成大顶堆）
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参数必须是一致的，不然会插入堆失败，最后一个元素还是在最后位置，导致插入失败
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<algorithm>
#include<cstring>
#include<vector>
using namespace std;

int main(){
    int n, v, total = 0;
    vector<int>h;
    make_heap(h.begin(), h.end(), greater<int>());
    //有了vector h  要想后面的操作首先要创建小根堆

    cin >> n;
    for (int i = 0; i < n;i++){
        cin >> v;
        h.push_back(v);//push_back之后可能会打乱
         push_heap(h.begin(), h.end(), greater<int> ());
         //所以就马上Push_heap  让他继续保持小根堆
    }
  while(h.size()>1){
      pop_heap(h.begin(),h.end(),greater<int>());
      //pop 想知道弹出后继续保持啥根堆，所以要传入跟当初make的时候一样的参数，也就是对处理堆顶元素，其他的做了一遍调整，因为下面想取出堆顶元素，所以他除外
      int v1=h.back();//取得堆顶元素
      h.pop_back();//抠出来，之前的pop_heap没对他影响

    pop_heap(h.begin(),h.end(),greater<int>());
      int v2=h.back();
      h.pop_back();
      
      h.push_back(v1+v2);//放进去
      push_heap(h.begin(),h.end(),greater<int>());//要算上+1，而你正好刚刚push进来一个

      total += (v1 + v2);
  }
  cout<<total;
        return 0;
}