【C++】priority_queue模拟实现过程中值得注意的点

樊梓慕：个人主页

 个人专栏：《C语言》《数据结构》《蓝桥杯试题》《LeetCode刷题笔记》《实训项目》《C++》《Linux》《算法》

每一个不曾起舞的日子，都是对生命的辜负

---

前言

本篇文章旨在记录博主在模拟实现priority_queue适配器中遇到的一些问题，希望与大家共勉。

---

欢迎大家收藏以便未来做题时可以快速找到思路，巧妙的方法可以事半功倍。

=========================================================================

GITEE相关代码：fanfei_c的仓库

=========================================================================

---

1.priority_queue的介绍

• priority_queue顾名思义：『 优先级队列』；
• 人话：堆。

priority_queue就是基于vector容器的堆，所以未来涉及到『 堆的应用』都可以使用priority_queue适配器。

默认为大堆。

同样通过模板参数的方式来决定底层容器和构建小堆。

比如：

• 使用『 vector』作为底层容器，内部构造『 大堆』结构。

priority_queue, less> q1;

• 使用『 vector』作为底层容器，内部构造『 小堆』结构。

priority_queue, greater> q2;

• 不指定底层容器和内部需要构造的堆结构，采用默认即vector、大堆。

priority_queue q;

比较奇怪的是：

构建大堆要传less，构建小堆要传greater，容易记混。

• 『 大堆』less就是逐渐变小；
• 『 小堆』greater就是逐渐变大。

---

2.堆的回顾

2.1向上调整算法

向上调整算法的前提是『 祖先是堆』。

以小堆为例：

1.给定向上调整的起点（孩子节点下标），根据起点下标计算双亲节点下标。

孩子节点与双亲结点间的下标关系：

• child=parent*2+1 || child=parent*2+2;
• parent=(child-1)/2;

2.比较孩子节点与双亲节点数值大小，若孩子节点小于于双亲节点，则交换两者，并将孩子节点的下标更新为之前的双亲节点下标，根据最新的孩子节点下标重新计算双亲节点下标，重复这一过程直到孩子节点为根节点。

//堆的向上调整（小堆）
void AdjustUp(vector& v, int child)
{
	int parent = (child - 1) / 2; //通过child计算parent的下标
	while (child > 0)//调整到根结点的位置截止
	{
		if (v[child] < v[parent])//孩子结点的值小于父结点的值
		{
			//将父结点与孩子结点交换
			swap(v[child], v[parent]);
			//继续向上进行调整
			child = parent;
			parent = (child - 1) / 2;
		}
		else
		{
			break;
		}
	}
}

---

2.2向下调整算法

向下调整算法的前提是『 左右子树是堆』。

以小堆为例：

1.给定向下调整的起点（双亲节点下标）和节点总数，根据起点下标计算孩子节点下标。

注意：向下调整时，若有两个孩子节点，则需要确保调整的是较小的孩子节点。

2.比较孩子节点与双亲节点数值大小，若孩子节点小于双亲节点，则交换两者，并将双亲节点的下标更新为之前的孩子节点下标，根据最新的双亲节点下标重新计算孩子节点下标，重复这一过程直到孩子节点超出节点总数。

//堆的向下调整（小堆）
void AdjustDown(vector& v, int n, int parent)
{
	//child记录左右孩子中值较大的孩子的下标
	int child = 2 * parent + 1;//先默认其左孩子的值较小
	while (child < n)
	{
		if (child + 1 < n && v[child+1] < v[child])//右孩子存在并且右孩子比左孩子还小
		{
			child = child + 1;//较小的孩子改为右孩子
		}
		if (v[child] < v[parent])//左右孩子中较小孩子的值比父结点还小
		{
			//将父结点与较小的子结点交换
			swap(v[child], v[parent]);
			//继续向下进行调整
			parent = child;
			child = 2 * parent + 1;
		}
		else
		{
			break;
		}
	}
}

---

3.构建大小堆的模板参数问题『仿函数 』 

这里同样涉及到模板参数的传递问题。

在学习堆时，我们知道可以通过改变比较方式从而实现建大堆或者建小堆：

但我们不能每次都手动的去改变这个符号从而满足用户需求。

所以这里我们同样可以利用『 仿函数』来解决这一问题。

仿函数是一个类或结构体，它重载了operator()运算符，使其可以像函数一样被调用。

仿函数的实例可以像函数指针一样传递给STL算法或容器的操作，从而实现自定义行为。

如同我们今天的例子，_com是less类的实例对象，less类重载了操作符()，使其类似于函数调用，内部实现比较大小的功能返回布尔值，_com()就是仿函数，该仿函数模拟了函数行为，使if的判断条件为真或假，从而达到我们>、<的目的。 

 

• 函数参数传参是在『 使用时传参』，传递的是『 对象』；
• 模板参数传参是在『 编译时传参』，传递的是『 类型』。

---

4.模拟实现源码

#pragma once

namespace F
{
    //比较方式：减小堆
	template
	class less
	{
	public:
		bool operator()(const T& x, const T& y)
		{
			return x < y;
		}
	};
    //比较方式：建大堆
	template
	class greater
	{
	public:
		bool operator()(const T& x, const T& y)
		{
			return x > y;
		}
	};
	template, class Compare = less>
	class priority_queue
	{
	public:
        //向上调整
		void adjust_up(int child)
		{
			int parent = (child - 1) / 2;
			while (child > 0)
			{
                //通过模板参数Compare确定是否需要交换结点位置
				if (_com(_con[child],_con[parent]))
				{
					swap(_con[child], _con[parent]);
					child = parent;
					parent = (child - 1) / 2;
				}
				else
				{
					break;
				}
			}
		}
        //向下调整
		void adjust_down(int parent)
		{
			int child = parent * 2 + 1;

			while (child < _con.size())
			{
				if (child + 1 < _con.size() && _com(_con[child+1],_con[child]))
				{
					++child;
				}

				if (_com(_con[child], _con[parent]))
				{
					swap(_con[child], _con[parent]);
					parent = child;
					child = parent * 2 + 1;
				}
				else
				{
					break;
				}
			}
		}

		void push(const T& x)
		{
			_con.push_back(x);

			adjust_up(_con.size() - 1);
		}

		void pop()
		{
			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;//底层容器
		Compare _com;//比较方式
	};
}

---

=========================================================================

如果你对该系列文章有兴趣的话，欢迎持续关注博主动态，博主会持续输出优质内容

博主很需要大家的支持，你的支持是我创作的不竭动力

~ 点赞收藏＋关注 ~

=========================================================================