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


     

    目录

    一、priority_queue的简介

    二、priority_queue的使用

    三、priority_queue的模拟实现

    1.基本框架的实现

    2.向上调整算法

    3.向下调整算法

    4.仿函数的介绍

    5.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是大堆。

    常用函数说明
    priority_queue( )默认构造函数
    priority_queue(first,last)迭代器区间构造
    empty( )检查优先级队列是否为空
    top( )返回优先级队列最大(最小)元素,即堆顶元素
    push( )在优先级队列中插入元素
    pop( )删除优先级队列中最大(最小)元素,即堆顶元素

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

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

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

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

    三、priority_queue的模拟实现

    1.基本框架的实现

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

    1. namespace mlg
    2. {
    3. template<class T, class Container = vector<T>>
    4. class priority_queue
    5. {
    6. public:
    7. //无参构造函数
    8. priority_queue() {}
    9. //尾插元素
    10. void push(const T& x)
    11. {
    12. _con.push_back(x);//调用所传容器的尾插
    13. }
    14. //移除堆顶的元素
    15. void pop()
    16. {
    17. assert(!_con.empty());
    18. swap(_con[0], _con[_con.size() - 1]);
    19. _con.pop_back();
    20. }
    21. //取堆顶的元素
    22. const T& top() {return _con[0];}
    23. //返回优先级队列有效元素的个数
    24. size_t size() {return _con.size();}
    25. //判断优先级队列是否为空
    26. bool empty() {return _con.empty();}
    27. private:
    28. Container _con;
    29. };
    30. }

    2.向上调整算法

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

     

    1. //向上调整
    2. void adjust_up(size_t child)
    3. {
    4. size_t parent = (child - 1) / 2;
    5. while (child > 0)
    6. {
    7. if (_con[child] > _con[parent])
    8. {
    9. swap(_con[child], _con[parent]);
    10. child = parent;
    11. parent = (child - 1) / 2;
    12. }
    13. else
    14. {
    15. break;//存在孩子已经小于父亲了,就不需要调整了
    16. }
    17. }
    18. }

    3.向下调整算法

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

      

    1. //向下调整
    2. void adjust_down(size_t parent)
    3. {
    4. size_t child = (parent * 2) + 1;
    5. while (child < _con.size())
    6. {
    7. //区分左右孩子谁大
    8. if (child + 1 < _con.size() && _con[child + 1] > _con[child])
    9. {
    10. ++child;
    11. }
    12. //比较最大的孩子与父亲的大小
    13. if (_con[child] > _con[parent])
    14. {
    15. swap(_con[child], _con[parent]);
    16. parent = child;
    17. child = parent * 2 + 1;
    18. }
    19. else
    20. {
    21. break;
    22. }
    23. }
    24. }

    4.仿函数的介绍

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

    举例: 

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

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

    1. template<class T>
    2. struct Less
    3. {
    4. bool operator()(const T& x, const T& y)
    5. {
    6. return x < y;
    7. }
    8. };
    9. template<class T>
    10. struct Greater
    11. {
    12. bool operator()(const T& x, const T& y)
    13. {
    14. return x > y;
    15. }
    16. };
    17. int main()
    18. {
    19. Less<int> le; //Less:仿函数类型 le:函数对象
    20. cout << le(1, 2) << endl;
    21. Greater<int> gt; //Greater:仿函数类型 gt:函数对象
    22. cout << gt(1, 2) << endl;
    23. return 0;
    24. }

    5.priority_queue完整实现

    1. namespace mlg
    2. {
    3. template<class T>
    4. struct Less
    5. {
    6. bool operator()(const T& x, const T& y)
    7. {
    8. return x < y;
    9. }
    10. };
    11. template<class T>
    12. struct Greater
    13. {
    14. bool operator()(const T& x, const T& y)
    15. {
    16. return x > y;
    17. }
    18. };
    19. //大的优先级高 --- 先实现大堆
    20. template<class T, class Container = vector<T>, class Compare = Less<T>>
    21. class priority_queue
    22. {
    23. private:
    24. //向上调整
    25. void adjust_up(size_t child)
    26. {
    27. Compare com;//定义了一个仿函数的函数对象,它可以通过上面两个类实现>和<的比较(相互交换)
    28. size_t parent = (child - 1) / 2;
    29. while (child > 0)
    30. {
    31. //if (_con[child] > _con[parent]) 等价于 if (_con[parent] < _con[child])
    32. if (com(_con[parent], _con[child]))
    33. {
    34. swap(_con[child], _con[parent]);
    35. child = parent;
    36. parent = (child - 1) / 2;
    37. }
    38. else
    39. {
    40. break;//存在孩子已经小于父亲了,就不需要调整了
    41. }
    42. }
    43. }
    44. //向下调整
    45. void adjust_down(size_t parent)
    46. {
    47. Compare com;
    48. size_t child = (parent * 2) + 1;
    49. while (child < _con.size())
    50. {
    51. //if (child + 1 < _con.size() && _con[child + 1] > _con[child])
    52. if (child + 1 < _con.size() && com(_con[child], _con[child + 1]))
    53. {
    54. ++child;
    55. }
    56. //if (_con[child] > _con[parent])
    57. if (com(_con[parent], _con[child]))
    58. {
    59. swap(_con[child], _con[parent]);
    60. parent = child;
    61. child = parent * 2 + 1;
    62. }
    63. else
    64. {
    65. break;
    66. }
    67. }
    68. }
    69. public:
    70. priority_queue()
    71. {}
    72. template <class InputIterator>
    73. priority_queue(InputIterator first, InputIterator last)
    74. :_con(first, last)
    75. {
    76. //建堆
    77. for (int i = (_con.size() - 1 - 1) / 2; i >= 0; --i)
    78. {
    79. adjust_down(i);
    80. }
    81. //在使用底部默认容器进行迭代器区间初始化时,我们需要先将整个堆构建成大堆或者小堆
    82. //我们可以倒着建堆,从第一个非叶子结点开始建堆
    83. }
    84. void push(const T& x)
    85. {
    86. _con.push_back(x);
    87. adjust_up(_con.size() - 1);//尾插数据需要进行向上调整,调整至大堆(小堆)
    88. }
    89. void pop()
    90. {
    91. assert(!_con.empty());
    92. swap(_con[0], _con[_con.size() - 1]);
    93. _con.pop_back();
    94. adjust_down(0);//交换头尾数据后,为确保堆的结构不变,需要进行向下调整
    95. }
    96. const T& top() {return _con[0];}
    97. size_t size() {return _con.size();}
    98. bool empty() {return _con.empty();}
    99. private:
    100. Container _con;
    101. };
    102. void test_priority_queue()
    103. {
    104. //priority_queue<int> pq; //测试Less --- 默认大的数优先级高
    105. priority_queue<int, vector<int>, Greater<int>> pq; //测试Greater --- 默认小的数优先级高
    106. pq.push(4);
    107. pq.push(3);
    108. pq.push(1);
    109. pq.push(8);
    110. pq.push(2);
    111. //判断谁的优先级高
    112. while (!pq.empty())
    113. {
    114. cout << pq.top() << " ";
    115. pq.pop();
    116. }
    117. cout << endl;
    118. }
    119. }
  • 相关阅读:
    1000亩盐碱地试验田 国稻种芯-田国庆:拓荒精神荒滩变良田
    mybatis-plus使用generator快速生成代码,并手动添加数据库命令
    MySQL高阶语句
    成功案例 | 安超云助力兰州大学第二医院搭建新型IT基础设施平台 提升医疗信息资源利用率
    2022年SQL大厂高频实战面试题(详细解析)
    【操作系统笔记】南京大学jyy老师
    mvc基于ASP NET大学生校园招聘网站的C#设计与开发
    【C++】DICOM医学影像工作站PACS源码
    Java · 数组 · 作为方法的参数 · 作为方法的返回值 · 二分查找 · 冒泡排序
    k8s的入门和项目部署
  • 原文地址:https://blog.csdn.net/sjsjnsjnn/article/details/125496716