使用priority_queue建立小顶推

详解priority_queue与小顶堆

最新推荐文章于 2024-09-28 19:27:15 发布

原创最新推荐文章于 2024-09-28 19:27:15 发布 · 2.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #优先队列 #小顶堆

C++ 专栏收录该内容

14 篇文章

订阅专栏

本文详细介绍了C++标准库中的priority_queue模板类及其使用方法，并通过具体示例展示了如何自定义比较函数来实现小顶堆。同时，文章还解释了less模板类的作用以及如何通过重载操作符来改变堆的类型。

template < 
   class Type,  
   class Container=vector<Type>, 
   class Compare=less<typename Container::value_type>  
> 
class priority_queue

注意，priority_queue是一个模板类，它的定义形式如上;其中，它的每一个参数也是一个类，这里我们主要解释一下Compare 参数。

其中，Compare 参数是一个类，里面的比较函数决定了堆是大顶堆还是小顶堆。一般而言, 默认是大顶堆。但是，在某些情况下，我们需要一个固定大小的小顶堆（比如要寻找某个集合中最大的两个元素。

现在，我们再看一下其中的less模板类：

template <class T> struct less {
  bool operator() (const T& x, const T& y) const {return x<y;}
  typedef T first_argument_type;
  typedef T second_argument_type;
  typedef bool result_type;
};

其中，里面最关键的是一个()操作符。

这里，我们给出一个例子：

struct Node{
    int x, y;
    Node( int a= 0, int b= 0 ):
        x(a), y(b) {}
};
 
struct cmp{
    bool operator() ( Node a, Node b ){
        if( a.x== b.x ) return a.y> b.y;
         
        return a.x> b.x; }
};
 
    priority_queue<Node, vector<Node>, cmp> q;

注意，这个堆是小顶堆。实际上，我们是按照operator ()的操作，将新添加的元素放在堆底的。