优先级队列
- 优先队列是一种容器适配器,根据严格的弱排序标准,它的第一个元素总是它所包含的元素中最大的。
- 此上下文类似于堆,在堆中可以随时插入元素,并且只能检索最大堆元素(优先队列中位于顶部的元素)。
- 优先队列被实现为容器适配器,容器适配器即将特定容器类封装作为其底层容器类,queue提供一组特定的成员函数来访问其元素。元素从特定容器的“尾部”弹出,其称为优先队列的顶部。
- 底层容器可以是任何标准容器类模板,也可以是其他特定设计的容器类。容器应该可以通过随机访问迭
代器访问,并支持以下操作:
empty():检测容器是否为空
size():返回容器中有效元素个数
front():返回容器中第一个元素的引用
push_back():在容器尾部插入元素
pop_back():删除容器尾部元素 - 标准容器类vector和deque满足这些需求。默认情况下,如果没有为特定的priority_queue类实例化指定容器类,则使用vector。
- 需要支持随机访问迭代器,以便始终在内部保持堆结构。容器适配器通过在需要时自动调用算法函数make_heap、push_heap和pop_heap来自动完成此操作。
优先级队列接口函数
构造函数
优先级队列默认使用vector作为其底层存储数据的容器,在vector上又使用了堆算法将vector中元素构造成堆的结构,因此priority_queue就是堆,所有需要用到堆的位置,都可以考虑使用priority_queue。注意:默认情况下priority_queue是大堆。
这里我们可以看到可以指定优先级队列所使用的容器以及比较方法,传入不同的比较方法可以得到大堆或者小堆。
代码测试如下:
void test_priotity_queue()
{
//priority_queue<int> pq;//默认数字大的优先级高,默认给的仿函数是less
priority_queue<int, vector<int>, greater<int>> pq;//控制小的优先级高
pq.push(3);
pq.push(1);
pq.push(2);
pq.push(5);
pq.push(8);
pq.push(7);
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
}
下面是不传入比较的函数,默认是形成大堆(从大到小)
void test_priotity_queue()
{
//priority_queue<int> pq;//默认数字大的优先级高,默认给的仿函数是less
priority_queue<int, vector<int>> pq;//控制小的优先级高
pq.push(3);
pq.push(1);
pq.push(2);
pq.push(5);
pq.push(8);
pq.push(7);
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
}
那么这种比较函数是怎么设计的呢?这就涉及到仿函数的功能了。
仿函数
仿函数实际上是一个()的重载,他通过模板来定义一个类,类中进行()重载,从而可以让对象像函数一样去使用,实现类似于函数的功能,仿函数实现如下:
template<class T>
struct Less//仿函数
{
bool operator()(const T& x, const T& y) const
{
return x < y;
}
};
template<class T>
struct Greater//仿函数,可以让对象像函数一样去使用,实际上用的是()的重载
{
bool operator()(const T& x, const T& y) const
{
return x > y;
}
};
empty
这个接口函数很简单,就是判断优先级队列是否为空。测试代码如下:
void test_priotity_queue3()
{
priority_queue<int> pq;
cout << pq.empty() << endl;
pq.push(1);
pq.push(2);
pq.push(3);
cout << pq.empty() << endl;
}
size
返回优先级队列的大小
void test_priotity_queue3()
{
priority_queue<int> pq;
pq.push(1);
pq.push(2);
pq.push(3);
cout << pq.size() << endl;
}
top
返回堆顶的元素
void test_priotity_queue()
{
priority_queue<int, vector<int>, greater<int>> pq;//控制小的优先级高
pq.push(3);
pq.push(1);
pq.push(2);
pq.push(5);
pq.push(8);
pq.push(7);
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
}
建立小堆,每次取出最小的元素
push/pop
push和pop功能是往堆里插入和删除数据,往堆里插入和删除数据需要进行数据的挪动,注意这点即可,后面模拟实现会介绍插入和删除的做法,这里的功能代码案例可以参考上面的top。
模拟实现优先级队列
大体框架
优先级队列和stack/queue类似,都直接使用了容器,这里优先级队列默认使用的是vector容器,因此可以这样定义:优先级队列的构造函数可以不用自己写默认使用其他容器的即可,析构函数也是一样,使用默认生成的即可。
template<class T, class Container = vector<T>,class Compare = Less<T>>
class priority_queue
{
priority_queue()
{
}
private:
Container _con;
};
向上调整
优先级队列一般在插入数据的时候,需要进行向上调整,从自己开始每次和自己的parent比较,如果满足比较条件则停止,不满足则进行交换,然后继续向上,直到堆顶。
void adjust_up(size_t child)
{
Compare cmp;
size_t parent = (child - 1) / 2;
while (child > 0)
{
if (cmp(_con[parent] , _con[child]))
{
std::swap(_con[child], _con[parent]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
向下调整
向下调整和向上调整类似,都是为了堆中数据的有序性存在的,向下调整一般多用于删除数据的时候,调整原则和向上类似,都是parent和child的比较,但是向下调整需要注意一下还需要比较左孩子和右孩子哪个大/小。
void adjust_down(size_t parent)
{
Compare cmp;
size_t child = (parent * 2) + 1;
while (child < _con.size())
{
if (child + 1 < _con.size() && cmp(_con[child] , _con[child+1]))
{
++child;
}
if (cmp(_con[parent] , _con[child]))
{
std::swap(_con[child], _con[parent]);
parent = child;
child = (parent * 2) + 1;
}
else
{
break;
}
}
}
迭代器构造函数
这里的使用迭代器构造函数,直接使用vector等容器进行初始化,然后从倒数第二层的叶子节点开始,依次向下调整来建堆。
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);
}
}
push/pop
这里的插入删除操作在实现了向上/向下调整后,很容易就可以实现。插入操作就是先直接插入数据,然后让该数据开始向上依次比较,进行向上调整。删除操作则是将堆顶和最后一个元素交换,然后删除最后一个元素,在让交换后的堆顶的元素开始向下调整,保持堆的有序性即可。
void push(const T& x)
{
_con.push_back(x);
adjust_up(_con.size() - 1);
}
void pop()
{
assert(!_con.empty());
std::swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
adjust_down(0);
}
其他功能
top接口直接返回堆顶的元素即可,size和empty可以直接使用容器自带的函数接口,复用即可。
const T& top()
{
return _con[0];
}
size_t size()
{
return _con.size();
}
bool empty()
{
return _con.empty();
}