优先队列
- 前言
- 优先队列
- 仿函数
- 头文件
前言
本篇主要讲解优先队列及其底层实现。
优先队列
优先队列的本质就是个堆,其与queue一样,都是容器适配器,不过优先队列是默认为vector实现的。priority_queue的接口优先队列默认为大根堆。
仿函数
我们观看文档可以发现
优先队列是有三个参数的,第二个参数即默认用vector进行实现优先队列,第三个参数即默认为大堆,less是已经实现了的仿函数,这里注意的是大堆是传的less,小堆传的是greater,是反着的。
仿函数实际是一个类,类中重载了()这个运算符,仿函数的实现能让我们自己定义来比较的标准,而不是用默认的。例如将默认为大堆更改为小堆,将sort默认为升序改为降序。
template<class T>
class less
{
bool operator()(const T& a1, const T& a2)
{
return a1 < a2;
}
};
头文件
#include<iostream>
#include<vector>
#include<algorithm>
namespace prime
{
//这是仿函数,仿函数是一个类
template<class T>
class less
{
public:
bool operator()(const T& a1, const T& a2)
{
return a1 < a2;
}
};
template<class T>
class greater
{
public:
bool operator()(const T& a1, const T& a2)
{
return a1 > a2;
}
};
template<class T>
class less<T*>//特化
{
public:
bool operator()(const T* const & x, const T* const & y)
{
return *x < *y;
}
};
template<class T,class container = vector<T>,class cmp = less<T>>
class priority_queue
{
public:
priority_queue()
{}
void push_back(const T& x)
{
_con.push_back(x);
adjustup(_con.size() - 1);
}
const T& top()
{
return _con[0];
}
void pop()
{
swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
adjustdown(0);
}
bool empty()
{
return _con.empty();
}
size_t size()
{
return _con.size();
}
private:
void adjustup(int child)
{
cmp com;//需要先实例化出一个com对象
while (child > 0)
{
int parent = (child - 1) / 2;
//if (_con[parent] < _con[child])
//if(com(_con[parent],_con[child]))
if(cmp()(_con[parent],_con[child]))//匿名对象
{
swap(_con[child], _con[parent]);
child = parent;
}
else
break;
}
}
void adjustdown(int parent)
{
cmp com;
int child = parent * 2 + 1;
while (child < _con.size())
{
//if (child + 1 < _con.size() && _con[child] < _con[child + 1])
if (child + 1 < _con.size() && com(_con[child],_con[child + 1]))
{
child++;
}
//if (_con[parent] < _con[child])
if (com(_con[parent], _con[child]))
{
swap(_con[parent], _con[child]);
parent = child;
child = parent * 2 + 1;
}
else
break;
}
}
private:
container _con;
};
}