C++初阶学习——探索STL奥秘——标准库中的priority_queue与模拟实现

Joker100852024-09-21 16:01

1.priority_queque的介绍

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

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

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

  3. 底层容器可以是任何标准容器类模板，也可以是其他特定设计的容器类。

容器应该可以通过随机访问迭代器访问，并支持以下操作：

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

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

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

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

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

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

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

2.priority_queue的基本使用

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

注意： 默认情况下priority_queue是大堆。

2.1构造函数

优先级队列的构造方式有两种：直接构造一个空对象 和 通过迭代器区间进行构造

cpp 复制代码 
#include <iostream>
#include <vector>
#include <queue>	//注意：优先级队列包含在 queue 的头文件中

using namespace std;

int main()
{
	priority_queue<int> pq;	//直接构造一个空对象，默认为大堆
	cout << typeid(pq).name() << endl;	//查看类型
	return 0;
}

注意默认仿函数为less，这个less决定了数据默认建大堆

通过迭代器区间构造对象

cpp 复制代码 
#include <iostream>
#include <vector>
#include <queue>	//注意：优先级队列包含在 queue 的头文件中

using namespace std;

int main()
{
	vector<char> vc = { 'a','b','c','d','e' };
	priority_queue<char, deque<char>, greater<char>> pq(vc.begin(), vc.end());	//现在是小堆
	cout << typeid(pq).name() << endl;	//查看类型
	cout << "==========================" << endl;
	while (!pq.empty())
	{
		//将小堆中的堆顶元素，依次打印
		cout << pq.top() << " ";
		pq.pop();
	}
	return 0;
}

注意:

将比较方式改为 greater 后，生成的是小堆

并且如果想修改比较方式的话，需要指明模板参数2 底层容器

因为比较方式位于模板参数3，不能跳跃缺省(遵循缺省参数规则)

2.2成员函数

cpp 复制代码 
#include <iostream>
#include <vector>
#include <queue>	//注意：优先级队列包含在 queue 的头文件中

using namespace std;

void Print(const priority_queue<int>& pq)
{
	cout << "是否为空：" << pq.empty() << endl;
	cout << "堆中的有效元素个数：" << pq.size() << endl;
	cout << "堆顶元素：" << pq.top() << endl;
	cout << "=================" << endl;
}

int main()
{
	vector<int> v = { 27,15,19,18,28,34,65,49,25,37 };
	priority_queue<int> pq(v.begin(), v.end());	//默认生成大堆
	Print(pq);

	pq.push(10);
	pq.push(100);
	Print(pq);

	pq.pop();
	pq.pop();
	pq.pop();
	Print(pq);

	return 0;
}

2.3练习题

思路:

利用数组建立大小为 k 的小堆，将剩余数据与堆顶值比较，如果大于，就入堆

为什么建小堆?因为此时需要的是最大的值，建大堆可能会导致次大的值无法入堆

答案：

cpp 复制代码 
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int,vector<int>,greater<int>> q(nums.begin(),nums.begin()+k);
        for(int i=k;i<nums.size();i++)
        {
            if(q.top()<nums[i])
            {
                q.pop();
                q.push(nums[i]);       
            }
        }
        return q.top();
    }
};

3.模拟实现优先级队列

3.1构造函数

优先级队列 priority_queue 属于容器适配器的一种，像栈和队列一样，没有迭代器

同时也不需要实现自己的具体功能，调用底层容器的功能就行了

不过因为堆比较特殊，需要具备 向上调整 和向下调整 的能力，确保符合堆的规则

首先是基本框架

cpp 复制代码 
namespace bit
{

	template<class T,class Comtainer=vector<T> >
	class Priority_Queue
	{
	public:
		Priority_Queue() {}
		template<class InterIterator>
	private:
		Comtainer _con;
	};
}

接着是迭代器区间的默认构造

这个默认构造的功能是传递一段数据的迭代器区间给优先级队列，然后优先级队列会自动在内部将数据排列成大堆或者小堆

cpp 复制代码 
namespace bit
{

	template<class T,class Comtainer=vector<T> >
	class Priority_Queue
	{
	public:
		Priority_Queue() {}
		template<class InterIterator>
		Priority_Queue(InterIterator first,InterIterator end)
		{
			while (first != end)
			{
				_con.push_back(*first);/*将数据推送到容器内部*/
				first++;
			}
			for (int i = (_con.size() - 1 - 1) / 2; i >= 0; i--)
			{
				Adjust_Down(i);
			}
		}
	private:
		Comtainer _con;
	};
}

3.2成员函数

cpp 复制代码 
		bool empty()const/*判断是否为空：复用底层结构的判空函数*/
		{
			return _con.empty();
		}
		const T& top()const/*获取堆顶元素：堆顶元素即第一个元素（二叉树的祖宗）

*/
		{
			return _con[0];
		}
		size_t size()const/*获取优先级队列大小：复用获取大小的函数*/
		{
			return _con.size();
		}

注意: 以上三个函数均为涉及对象内容的改变，因此均使用 const 修饰 this 指针所指向的内容

在插入/删除数据后，需要确保堆能符合要求

大堆:父节点比子节点大

小堆:父节点比子节点小

因此每进行一次数据修改相关操作，都需要检查当前堆结构是否被破坏，这一过程称为调整

插入数据:尾插数据，然后向上调整

cpp 复制代码 
    void Adjust_Up(int child)
{
	int parent = child / 2 ;
	while(child>0)
	if (_con[parent] < _con[child])
	{
		std::swap(_con[parent], _con[child]);
		parent = child;
		child = parent-1/2;
	}
	else
		break;
}

     void push(const T& x)
	{
		_con.push_back(x);
		Adjust_Up(_con.size()-1);
	}

删除数据:将堆顶数据交换至堆底，删除堆底元素，再向下调整堆

这步至关重要，pop的时候将堆顶数据交换至堆底，再通过容器的成员函数pop_back删除堆底的元素，也就是删除掉容器存储的数据中最大或者最小的元素 ，再从堆顶向下调整数据可以重新再这段数据中筛选出最大或者最小的元素推送到堆顶

cpp 复制代码 
		void Adjust_Down(int parent)
		{
			int child = parent * 2 + 1;
			while (child < _con.size())
			{
				if (child+1<_con.size()&_con[child] <_con[child + 1])
				{
					child++;
				}
				if (_con[parent] <_con[child])
				{
					std::swap(_con[parent], _con[child]);
					parent = child;
					child = parent * 2 + 1;
				}
				else
					break;
			}
		}
        void pop()const
       {
	        std::swap(_con[0], _con[_con.size() - 1]);
	        _con.pop_back();
	        Adjust_Down(0);
       }

测试：

3.3仿函数

仿函数又名函数对象 function obiects

需要调用库#include<functional>

仿函数的主要作用是

借助类和运算符重载，做到同一格式兼容所有函数

这有点像函数指针

相比于函数指针又长又难理解的定义，仿函数的使用可谓是很简单了

下面是两个仿函数，作用是比较大小

cpp 复制代码 
	template<class T>
	class Less
	{
	public:
		bool operator()(const T& a, const T& b)
		{
			return a < b;
		}
	};
	template<class T>
	class Greater
	{
	public:
		bool operator()(const T& a, const T& b)
		{
			return a > b;
		}
	};

此时 priority_queue 中的模板参数升级为3个，而参数3的缺省值就是Less

cpp 复制代码 
template<class T,class Comtainer=vector<T>,class Compare=Greater<T> >

像这样，用仿函数生成的对象，将所有要用到比较大小的代码进行替换

cpp 复制代码 
Compare greater;
void Adjust_Down(int parent)
{
	int child = parent * 2 + 1;
	while (child < _con.size())
	{
		if (child+1<_con.size()&&greater(_con[child] , _con[child + 1]))
		{
			child++;
		}
		if (greater(_con[parent] , _con[child]))
		{
			std::swap(_con[parent], _con[child]);
			parent = child;
			child = parent * 2 + 1;
		}
		else
			break;
	}
}

这样就能实现排序顺序的自由控制

3.4特殊情况

在日期类中

cpp 复制代码 
class Date
{
public:
	Date(int year = 1970, int month = 1, int day = 1)
		: _year(year)
		, _month(month)
		, _day(day)
	{}
	bool operator<(const Date& d)const
	{
		return (_year < d._year) ||
			(_year == d._year && _month < d._month) ||
			(_year == d._year && _month == d._month && _day < d._day);
	}
	bool operator>(const Date& d)const
	{
		return (_year > d._year) ||
			(_year == d._year && _month > d._month) ||
			(_year == d._year && _month == d._month && _day > d._day);
	}
	friend std::ostream& operator<<(std::ostream& _cout, const Date& d)
	{
		_cout << d._year << "-" << d._month << "-" << d._day;
		return _cout;
	}
private:
	int _year;
	int _month;
	int _day;
};

此时会随机排序，因为我们传入的是指针 ，地址是随机的所以结果也是随机的

所以这里我们专门为它构造一个仿函数就可以了

cpp 复制代码 
//小于
template<class T>
struct LessPDate
{
	bool operator()(const T& p1, const T& p2)
	{
		return (*p1) < (*p2);
	}
};

//大于
template<class T>
struct GreaterPDate
{
	bool operator()(const T& p1, const T& p2)
	{
		return (*p1) > (*p2);
	}
};

4.源码

源码链接

相关推荐
AI街潜水的八角9 分钟前
基于C++的决策树C4.5机器学习算法（不调包）
c++·算法·决策树·机器学习
JSU_曾是此间年少39 分钟前
数据结构——线性表与链表
数据结构·c++·算法
此生只爱蛋2 小时前
【手撕排序2】快速排序
c语言·c++·算法·排序算法
何曾参静谧2 小时前
「C/C++」C/C++ 指针篇 之 指针运算
c语言·开发语言·c++
lulu_gh_yu3 小时前
数据结构之排序补充
c语言·开发语言·数据结构·c++·学习·算法·排序算法
ULTRA??3 小时前
C加加中的结构化绑定（解包，折叠展开）
开发语言·c++
凌云行者4 小时前
OpenGL入门005——使用Shader类管理着色器
c++·cmake·opengl
凌云行者4 小时前
OpenGL入门006——着色器在纹理混合中的应用
c++·cmake·opengl
～yY…s<#>4 小时前
【刷题17】最小栈、栈的压入弹出、逆波兰表达式
c语言·数据结构·c++·算法·leetcode
可均可可5 小时前
C++之OpenCV入门到提高004：Mat 对象的使用
c++·opencv·mat·imread·imwrite
热门推荐
01【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总02(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明03组基轨迹建模 GBTM的介绍与实现（Stata 或 R）04【AIGC】重塑未来的科技巨轮05全面解析：构建基于深度学习的安全帽检测系统（UI界面+YOLO代码+数据集）06【经验分享】Ubuntu22.04安装微信(linux官方版)07基于YOLOv10深度学习的CT扫描图像肾结石智能检测系统【python源码+Pyqt5界面+数据集+训练代码】深度学习实战、目标检测08Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO09RAG 实践- Ollama+RagFlow 部署本地知识库10【TC3xx芯片】TC3xx芯片电源管理系统PMS详解