【C++】STL——priority_queue的使用及模拟实现

简介: 【C++】STL——priority_queue的使用及模拟实现

一、priority_queue的简介

1. 优先级队列是一种容器适配器,根据严格的弱排序标准,它的第一个元素总是它所包含的元素中最大的。

2. 类似于堆,在堆中可以随时插入元素,并且只能检索最大堆元素(优先队列中位于顶部的元素)。

3. 优先队列被实现为容器适配器,容器适配器即将特定容器类封装作为其底层容器类queue提供一组特定的成员函数来访问其元素。元素从特定容器的“尾部”弹出,其称为优先队列的顶部。

4. 底层容器可以是任何标准容器类模板,也可以是其他特定设计的容器类。容器应该可以通过随机访问迭代器访问,并支持以下操作:

       empty():检测容器是否为空

       size():返回容器中有效元素个数

       front():返回容器中第一个元素的引用

       push_back():在容器尾部插入元素

       pop_back():删除容器尾部元素


5. 标准容器类vector和deque满足这些需求。默认情况下,如果没有为特定的priority_queue类实例化指定容器类,则使用vector。

6. 需要支持随机访问迭代器,以便始终在内部保持堆结构。容器适配器通过在需要时自动调用算法函数make_heap、push_heap和pop_heap来自动完成此操作。

二、priority_queue的使用

       优先级队列默认使用vector作为其底层存储数据的容器,在vector上又使用了 堆算法 将vector中元素构造成堆的结构,因此priority_queue就是堆,所有需要用到堆的位置,都可以考虑使用priority_queue。注意: 默认情况下priority_queue是大堆。

1ecd1b2606ed46e9956a89f231c9802c.png

image.png

优先级队列的模板参数有三个:

template<class T, class Container = vector<int>, class Compare = less<T>>;

       第一个template参数是元素类型,带有默认值的第二个template参数定义了priority_queue内部用来存放元素的容器,默认容器是vector。带有默认值的第三个template参数定义出“用以查找下一个最高优先级元素”的排序准则,默认以operator<作为比较标准(less),如果想以operator>作为比较标准可以定义为greater。(这里的第三个参数是仿函数)

//默认以vector作为底部容器,less作为比较准则(x < y)
priority_queue<int> pq;
//等价于priority_queue<int, vector<int>, Less<int>> pq;
//如果想要以(x > y)比较准则,可以显示给第三个模板参数传greater<int>,但是第二参数必须要写
priority_queue<int, vector<int>, greater<int>> pq;
#include <iostream>
#include <vector>
#include <queue>
#include <functional>
using namespace std;
void test_priority_queue2()
{
  //priority_queue<int> pq; //默认是大的优先级高 --- 默认给的仿函数是less
  //priority_queue<int, vector<int>, greater<int>> pq;  //如果想要控制小的优先级高 --- 给一个greater的仿函数(需要包functional头文件)
  priority_queue<int> pq1;
  pq1.push(1);
  pq1.push(2);
  vector<int> v1;
  v1.push_back(2);
  v1.push_back(3);
  v1.push_back(5);
  priority_queue<int> pq2(v1.begin(), v1.end());
  //判断(2 3 5)谁的优先级高
  while (!pq2.empty())
  {
    cout << pq2.top() << " ";//top相当于取的是堆顶的数据
    pq2.pop();
  }
  cout << endl;
}

三、priority_queue的模拟实现

1.基本框架的实现

       在实现时,我们先不实现第三个参数,仅仅实现前两个参数,这样一来整体难度就显得容易一些。优先级队列它也是队列,我们基本的函数要能够满足优先级队列的push、pop( )、top( )、size( )、empty( );基本上和之前的stack和queue一样;

namespace mlg
{
  template<class T, class Container = vector<T>>
  class priority_queue
  { 
  public:
        //无参构造函数
    priority_queue() {}
        //尾插元素
    void push(const T& x)
    {
      _con.push_back(x);//调用所传容器的尾插
    }
        //移除堆顶的元素
    void pop()
    {
      assert(!_con.empty());
      swap(_con[0], _con[_con.size() - 1]);
      _con.pop_back();
    }
        //取堆顶的元素
    const T& top() {return _con[0];}
        //返回优先级队列有效元素的个数
    size_t size() {return _con.size();}
        //判断优先级队列是否为空
    bool empty() {return _con.empty();}
  private:
    Container _con;
  };
}

2.向上调整算法

       优先级队列实际上是堆,在我们在基本框架的实现中,当我们push一个元素时,就需要对这个堆进行调整。因为优先级队列默认是大堆,在此我们实现出向上调整算法,将大的元素调整到最上面。

56e48d4eef43477daf52f0ec3a7b7150 (1).gif

//向上调整
void adjust_up(size_t child)
{
  size_t parent = (child - 1) / 2;
  while (child > 0)
  {
    if (_con[child] > _con[parent])
    {
      swap(_con[child], _con[parent]);
      child = parent;
      parent = (child - 1) / 2;
    }
    else
    {
      break;//存在孩子已经小于父亲了,就不需要调整了
    }
  }
}

3.向下调整算法

       对于基本框架中的pop函数是将堆顶的元素进行删除,但是我们并不能直接删除,如果直接删除会让堆的结构完全错乱,所以采用向下调整算法,在交换头尾数据后,再对这个堆的进行调整,让下一次取堆顶数据时也是最大的数据。

//向下调整
void adjust_down(size_t parent)
{
  size_t child = (parent * 2) + 1;
  while (child < _con.size())
  {
        //区分左右孩子谁大
    if (child + 1 < _con.size() && _con[child + 1] > _con[child])
    {
      ++child;
    }
        //比较最大的孩子与父亲的大小
    if (_con[child] > _con[parent])
    {
      swap(_con[child], _con[parent]);
      parent = child;
      child = parent * 2 + 1;
    }
    else
    {
      break;
    }
  }
}

4.仿函数的介绍

       仿函数:就是使一个类的使用看上去像一个函数。其实现就是在一个类中实现一个operator( ) ,这个类就具有了类似函数的行为,就是一个仿函数类了。

举例:

  我们应该记得C语言中的函数指针,它配合着回调函数来实现仿函数(可以看看qsort函数);


       在C++中,我们通过在一个类中重载括号运算符的方法使用一个函数对象(重载函数调用操作符的类,其对象常称为函数对象(function object),即它们是行为类似函数的对象。又称仿函数。)而不是一个普通函数。

template<class T>
struct Less
{
  bool operator()(const T& x, const T& y)
  {
    return x < y;
  }
};
template<class T>
struct Greater
{
  bool operator()(const T& x, const T& y)
  {
    return x > y;
  }
};
int main()
{
    Less<int> le; //Less:仿函数类型   le:函数对象
    cout << le(1, 2) << endl;
    Greater<int> gt; //Greater:仿函数类型   gt:函数对象
    cout << gt(1, 2) << endl;
    return 0;
}

5.priority_queue完整实现

namespace mlg
{
  template<class T>
  struct Less
  {
    bool operator()(const T& x, const T& y)
    {
      return x < y;
    }
  };
  template<class T>
  struct Greater
  {
    bool operator()(const T& x, const T& y)
    {
      return x > y;
    }
  };
  //大的优先级高 --- 先实现大堆
  template<class T, class Container = vector<T>, class Compare = Less<T>>
  class priority_queue
  { 
  private:
    //向上调整
    void adjust_up(size_t child)
    {
      Compare com;//定义了一个仿函数的函数对象,它可以通过上面两个类实现>和<的比较(相互交换)
      size_t parent = (child - 1) / 2;
      while (child > 0)
      {
        //if (_con[child] > _con[parent]) 等价于 if (_con[parent] < _con[child])
        if (com(_con[parent], _con[child]))
        {
          swap(_con[child], _con[parent]);
          child = parent;
          parent = (child - 1) / 2;
        }
        else
        {
          break;//存在孩子已经小于父亲了,就不需要调整了
        }
      }
    }
    //向下调整
    void adjust_down(size_t parent)
    {
      Compare com;
      size_t child = (parent * 2) + 1;
      while (child < _con.size())
      {
        //if (child + 1 < _con.size() && _con[child + 1] > _con[child])
        if (child + 1 < _con.size() && com(_con[child], _con[child + 1]))
        {
          ++child;
        }
        //if (_con[child] > _con[parent])
        if (com(_con[parent], _con[child]))
        {
          swap(_con[child], _con[parent]);
          parent = child;
          child = parent * 2 + 1;
        }
        else
        {
          break;
        }
      }
    }
  public:
    priority_queue()
    {}
    template <class InputIterator>
    priority_queue(InputIterator first, InputIterator last)
      :_con(first, last)
    {
      //建堆
      for (int i = (_con.size() - 1 - 1) / 2; i >= 0; --i)
      {
        adjust_down(i);
      }
            //在使用底部默认容器进行迭代器区间初始化时,我们需要先将整个堆构建成大堆或者小堆
            //我们可以倒着建堆,从第一个非叶子结点开始建堆
    }
    void push(const T& x)
    {
      _con.push_back(x);
      adjust_up(_con.size() - 1);//尾插数据需要进行向上调整,调整至大堆(小堆)
    }
    void pop()
    {
      assert(!_con.empty());
      swap(_con[0], _con[_con.size() - 1]);
      _con.pop_back();
      adjust_down(0);//交换头尾数据后,为确保堆的结构不变,需要进行向下调整
    }
    const T& top() {return _con[0];}
    size_t size() {return _con.size();}
    bool empty() {return _con.empty();}
  private:
    Container _con;
  };
  void test_priority_queue()
  {
        //priority_queue<int> pq; //测试Less --- 默认大的数优先级高
    priority_queue<int, vector<int>, Greater<int>> pq; //测试Greater --- 默认小的数优先级高
    pq.push(4);
    pq.push(3);
    pq.push(1);
    pq.push(8);
    pq.push(2);
    //判断谁的优先级高
    while (!pq.empty())
    {
      cout << pq.top() << " ";
      pq.pop();
    }
    cout << endl;
  }
}
目录
相关文章
|
22小时前
|
算法 编译器 C语言
从C语言到C++⑩(第四章_模板初阶+STL简介)如何学习STL(下)
从C语言到C++⑩(第四章_模板初阶+STL简介)如何学习STL
4 0
|
4天前
|
存储 算法 搜索推荐
C++|STL简介-string-vector基础运用
C++|STL简介-string-vector基础运用
|
6天前
|
设计模式 算法 C++
【C++】STL之迭代器介绍、原理、失效
【C++】STL之迭代器介绍、原理、失效
13 2
|
6天前
|
存储 C++ 容器
C++:STL - set & map
C++:STL - set & map
16 4
|
6天前
|
算法 安全 程序员
【C++】STL学习之旅——初识STL,认识string类
现在我正式开始学习STL,这让我期待好久了,一想到不用手撕链表,手搓堆栈,心里非常爽
17 0
|
6天前
|
存储 Serverless C++
【C++入门到精通】哈希 (STL) _ unordered_map _ unordered_set [ C++入门 ]
【C++入门到精通】哈希 (STL) _ unordered_map _ unordered_set [ C++入门 ]
11 1
|
6天前
|
存储 设计模式 算法
【C++/STL】stack和queue(容器适配器、优先队列、双端队列)
【C++/STL】stack和queue(容器适配器、优先队列、双端队列)
16 1
|
6天前
|
存储 编译器 C++
【C++/STL】list(常见接口、模拟实现、反向迭代器、)
【C++/STL】list(常见接口、模拟实现、反向迭代器、)
5 0
|
6天前
|
算法 C++ 容器
【C++/STL】vector(常见接口、模拟实现、迭代器失效)
【C++/STL】vector(常见接口、模拟实现、迭代器失效)
12 0
|
6天前
|
存储 算法 C++
详解C++中的STL(标准模板库)容器
【4月更文挑战第30天】C++ STL容器包括序列容器(如`vector`、`list`、`deque`、`forward_list`、`array`和`string`)、关联容器(如`set`、`multiset`、`map`和`multimap`)和容器适配器(如`stack`、`queue`和`priority_queue`)。它们为动态数组、链表、栈、队列、集合和映射等数据结构提供了高效实现。选择合适的容器类型可优化性能,满足不同编程需求。