模拟实现 priority_queue
基本实现思路
在优先级队列中,插入数据和删除数据的时间复杂度为 。
默认情况下的优先级队列是大堆,我们先不考虑用仿函数去实现兼容大堆小队排列问题,
我们先去实现大堆,先把基本的功能实现好,带着讲解完仿函数后再去进行优化实现。
优先级队列相较于普通的队列,其区别主要是在 push 和 pop 上,
即需要在插入 / 删除数据的同时,增添调整的功能,并且 STL 的优先级队列是由堆来维护的:
void push(const T& x) {
_con.push_back(x);
AdjustUp(_con.size() - 1);
}
入队时将数据推入后,从最后一个元素开始进行上调。
出队时采用堆的删除手法,将根元素(即队头)和最后一个元素进行交换,并调用尾删掉队头。
既然用了头尾交换法,我们就不得不对队列重新调整,所以从根位置进行下调,即可完成出队。
void pop() {
assert(!_con.empty());
swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
AdjustDown(0);
}
既然需要上调和下调操作,我们不可避免地需要实现上调和下调的算法。
我们这里暂时写死,设计成大堆的 AdjustUp 和 AdjustDown:
/* 向上调整算法 */
void AdjustUp(size_t child) {
size_t father = (child - 1) / 2;
while (child > 0) {
if (_con[father] < _con[child]) {
swap(_con[father], _con[child]);
child = father;
father = (child - 1) / 2;
}
else {
break;
}
}
}
/* 向下调整算法 */
void AdjustDown(size_t father) {
size_t child = father * 2 + 1; // 默认认为左孩子大
while (child < _con.size()) {
// 判断右孩子是否存在,并且是否比左孩子大?
if (child + 1 < _con.size() && _con[child] < _con[child + 1]) {
child += 1;
}
if (_con[father] < _con[child]) {
swap(_con[father], _con[child]);
father = child;
child = father * 2 + 1;
}
else {
break;
}
}
}
向上调整算法和向下调整算法实现完后,我们把剩下的可以直接用 _con 转换的接口实现一下:
const T& top() {
return _con[0];
}
size_t size() {
return _con.size();
}
bool empty() {
return _con.empty();
}
很简单,不过值得注意的是,返回堆顶数据的 top 接口,我们用了 const 引用返回。
思考:为什么这里要用引用?为什么还要加个 const?
① 考虑到如果这里 T 是 string,甚至是更大的对象,传值返回就有一次拷贝构造,代价就大了。
② 如果能让你随随便便修改我堆顶的数据,那岂不是 显得我很随便? 很危险?
大堆优先级队列
代码:大堆优先级队列 demo
#include <iostream>
#include <vector>
#include <assert.h>
using namespace std;
namespace foxny {
template<class T, class Container = vector<T>>
class priority_queue {
public:
/* 向上调整算法 */
void AdjustUp(size_t child) {
size_t father = (child - 1) / 2;
while (child > 0) {
if (_con[father] < _con[child]) {
swap(_con[father], _con[child]);
child = father;
father = (child - 1) / 2;
}
else {
break;
}
}
}
/* 向下调整算法 */
void AdjustDown(size_t father) {
size_t child = father * 2 + 1; // 默认认为左孩子大
while (child < _con.size()) {
// 判断右孩子是否存在,并且是否比左孩子大?
if (child + 1 < _con.size() && _con[child] < _con[child + 1]) {
child += 1;
}
if (_con[father] < _con[child]) {
swap(_con[father], _con[child]);
father = child;
child = father * 2 + 1;
}
else {
break;
}
}
}
/* 入数据 */
void push(const T& x) {
_con.push_back(x);
AdjustUp(_con.size() - 1);
}
/* 出数据 */
void pop() {
assert(!_con.empty());
swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
AdjustDown(0);
}
/* 返回堆顶数据 */
const T& top() {
return _con[0];
}
/* 返回大小 */
size_t size() {
return _con.size();
}
/* 判断是否为空 */
bool empty() {
return _con.empty();
}
private:
Container _con;
};
}
测试:目前我们是写死了的大堆,还没有加入仿函数,我们先跑一下看看
void test_priority_queue1() {
priority_queue<int> pQ;
pQ.push(2);
pQ.push(5);
pQ.push(1);
pQ.push(6);
pQ.push(8);
while (!pQ.empty()) {
cout << pQ.top() << " ";
pQ.pop();
}
cout << endl;
}
运行结果:
我们写死的最大问题是 —— 如果想按升序排列,就没办法排了。
我们这里写的用来排降序的大堆,如果想排升序我们需要写小堆,
C++ 在这里有一种叫仿函数的东西,可以控制这些东西,我们下面先来介绍一下仿函数。
仿函数(functor)
概念:使一个类的使用看上去像一个函数。可以像函数一样使用的对象,称为函数对象。
其实现就是在类中重载 operator(),使得该类具备类似函数的行为,就是一个仿函数类了。
C语言优先级,() 圆括号使用形式为 表达式 或 "作为函数形参列表的括号"
我们这里重载的其实就是:函数名(形参表)
代码:我们来写一个最简单的仿函数
namespace foxny {
struct less {
// 仿函数(函数对象)———— 对象可以像调用函数一样去使用
bool operator()(int x, int y) {
return x < y;
}
};
void test_functor() {
less lessCmp; // 定义一个对象(这里我用小驼峰,能看上去更像函数)
cout << lessCmp(10, 20) << endl; // 是对象,但是却能像函数一样去使用
}
}
运行结果:1
这里定义对象的时候我用的我是小驼峰,这样看上去更像是一个函数了,以假乱真。
我们定义的 lessCmp 是一个对象,但是我们却可以像使用函数一样给它传递参数:
lessCmp(10, 20);
我们还可以使其能够支持泛型,我们加一个 template 模板:
// 支持泛型
template<class T> struct less {
bool operator()(const T& x, const T& y) const {
return x < y;
}
};
void test_functor() {
less<int> lessCmp;
cout << lessCmp(10, 20) << endl;
}
less 是用来比较谁小的,对应的,我们再实现一个比较谁大的 greater:
// less: 小于的比较
template<class T> struct less {
bool operator()(const T& x, const T& y) const {
return x < y;
}
};
// greater: 大于的比较
template<class T> struct greater {
bool operator()(const T& x, const T& y) const {
return x > y;
}
};
void test_functor() {
less<int> lessCmp;
cout << lessCmp(10, 20) << endl;
greater<int> GreaterCmp;
cout << GreaterCmp(10, 20) << endl;
}
运行结果:1 0
一个对象能像函数一样用,看上去很神奇,实际上调用的只是我们重载的 operator() 而已。
这些操作都是归功于运算符重载功能,这就是仿函数!
思考:我们在C阶段不是学过函数指针么,用函数指针不香吗?
函数指针确实可以做到这些功能。但是在这里函数指针跟仿函数比,一点都不香。
在 C++ 里其实是非常鄙夷使用函数指针的,函数的指针在一些地方用起来非常复杂。
static void( * set_malloc_handler(void (*f)()))() {
void (* old)() = __malloc_alloc_oom_handler;
__malloc_alloc_oom_handler = f;
return(old);
}
这个函数的参数是什么?函数的返回值是什么?
这就很阴间了,这就是函数指针的杰作…… 所以 C++ 搞出了仿函数,简化了好多点。
加了仿函数后的优先级队列
我们现在用仿函数去比较那些数据,这样就不会写死了。
代码:priority_queue
#include <iostream>
#include <vector>
#include <functional> // 这里放的是一些仿函数
#include <assert.h>
using namespace std;
namespace foxny {
// less: 小于的比较
template<class T>
struct less {
bool operator()(const T& x, const T& y) const {
return x < y;
}
};
// greater: 大于的比较
template<class T>
struct greater {
bool operator()(const T& x, const T& y) const {
return x > y;
}
};
template<class T, class Container = vector<T>, class Compare = less<T>>
class priority_queue {
public:
/* 向上调整算法 */
void AdjustUp(size_t child) {
Compare cmpFunc;
size_t father = (child - 1) / 2;
while (child > 0) {
// if (_con[father] < _con[child]) {
if (cmpFunc(_con[father], _con[child])) {
swap(_con[father], _con[child]);
child = father;
father = (child - 1) / 2;
}
else {
break;
}
}
}
/* 向下调整算法 */
void AdjustDown(size_t father) {
Compare cmpFunc;
size_t child = father * 2 + 1; // 默认认为左孩子大
while (child < _con.size()) {
// if (child + 1 < _con.size() && _con[child] < _con[child + 1]) {
if (child + 1 < _con.size() && cmpFunc(_con[child], _con[child + 1])) {
child += 1;
}
// if (_con[father] < _con[child]) {
if (cmpFunc(_con[father], _con[child])) {
swap(_con[father], _con[child]);
father = child;
child = father * 2 + 1;
}
else {
break;
}
}
}
/* 入数据 */
void push(const T& x) {
_con.push_back(x);
AdjustUp(_con.size() - 1);
}
/* 出数据 */
void pop() {
assert(!_con.empty());
swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
AdjustDown(0);
}
/* 返回堆顶数据 */
const T& top() {
return _con[0];
}
/* 返回大小 */
size_t size() {
return _con.size();
}
/* 判断是否为空 */
bool empty() {
return _con.empty();
}
private:
Container _con;
};
}
测试:我们来测试一下效果如何:
// 大于比较,搞小堆
void test_priority_queue2() {
priority_queue<int, vector<int>, greater<int>> pQ;
pQ.push(2);
pQ.push(5);
pQ.push(1);
pQ.push(6);
pQ.push(8);
while (!pQ.empty()) {
cout << pQ.top() << " ";
pQ.pop();
}
cout << endl;
}
运行结果:1 2 4 6 8
// 小于比较,搞大堆
void test_priority_queue1() {
priority_queue<int> pQ;
pQ.push(2);
pQ.push(5);
pQ.push(1);
pQ.push(6);
pQ.push(8);
while (!pQ.empty()) {
cout << pQ.top() << " ";
pQ.pop();
}
cout << endl;
}
运行结果:8 6 5 2 1
这里的仿函数在编译器视角是这样的:
if (cmpFunc(_con[father], _con[child]))
👇
if (cmpFunc.operator()(_con[father], _con[child]))
👇
// 以 less 为例
template<class T>
struct less {
bool operator()(const T& x, const T& y) const {
return x < y;
}
};
最后返回 _con[father] < _con[child] ,true 还是 false
迭代器的实现
// 创造空的优先级队列
priority_queue()
: _con()
{}
// 迭代器
template<class InputIterator>
priority_queue (InputIterator first, InputIterator last)
: _con(first, last)
{
while (first != last) {
_con.push_back(*first++);
}
for (int father = (_con.size()-1-1) / 2; father >= 0; father++) {
AdjustDown(father);
}
}
我们就可以拿这个来解决一些问题,比如:
void test_priority_queue3() {
// TOP K
int arr[] = {1,4,2,7,8,9};
priority_queue<int> pQ(arr, arr + 6); // 传一个迭代器区间
}
完整代码
#include <iostream>
#include <vector>
#include <functional> // 这里放的是一些仿函数
#include <assert.h>
using namespace std;
namespace foxny {
// less: 小于的比较
template<class T>
struct less {
bool operator()(const T& x, const T& y) const {
return x < y;
}
};
// greater: 大于的比较
template<class T>
struct greater {
bool operator()(const T& x, const T& y) const {
return x > y;
}
};
template<class T, class Container = vector<T>, class Compare = less<T>>
class priority_queue {
Compare _cmpFunc;
public:
// 创造空的优先级队列
priority_queue()
: _con()
{}
// 迭代器
template<class InputIterator>
priority_queue (InputIterator first, InputIterator last)
: _con(first, last)
{
int count = _con.size();
int root = ((count - 2) >> 1);
for (; root >= 0; root--) {
AdjustDown(root);
}
}
/* 向上调整算法 */
void AdjustUp(size_t child) {
Compare cmpFunc;
size_t father = (child - 1) / 2;
while (child > 0) {
// if (_con[father] < _con[child]) {
if (cmpFunc(_con[father], _con[child])) {
swap(_con[father], _con[child]);
child = father;
father = (child - 1) / 2;
}
else {
break;
}
}
}
/* 向下调整算法 */
void AdjustDown(size_t father) {
Compare cmpFunc;
size_t child = father * 2 + 1; // 默认认为左孩子大
while (child < _con.size()) {
// if (child + 1 < _con.size() && _con[child] < _con[child + 1]) {
if (child + 1 < _con.size() && cmpFunc(_con[child], _con[child + 1])) {
child += 1;
}
// if (_con[father] < _con[child]) {
if (cmpFunc(_con[father], _con[child])) {
swap(_con[father], _con[child]);
father = child;
child = father * 2 + 1;
}
else {
break;
}
}
}
/* 入数据 */
void push(const T& x) {
_con.push_back(x);
AdjustUp(_con.size() - 1);
}
/* 出数据 */
void pop() {
assert(!_con.empty());
swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
AdjustDown(0);
}
/* 返回堆顶数据 */
const T& top() {
return _con[0];
}
/* 返回大小 */
size_t size() {
return _con.size();
}
/* 判断是否为空 */
bool empty() {
return _con.empty();
}
private:
Container _con;
};
// 小于比较,搞大堆
void test_priority_queue1() {
priority_queue<int> pQ;
pQ.push(2);
pQ.push(5);
pQ.push(1);
pQ.push(6);
pQ.push(8);
while (!pQ.empty()) {
cout << pQ.top() << " ";
pQ.pop();
}
cout << endl;
}
// 大于比较,搞小堆
void test_priority_queue2() {
priority_queue<int, vector<int>, greater<int>> pQ;
pQ.push(2);
pQ.push(5);
pQ.push(1);
pQ.push(6);
pQ.push(8);
while (!pQ.empty()) {
cout << pQ.top() << " ";
pQ.pop();
}
cout << endl;
}
}
int main(void)
{
foxny::test_priority_queue1();
foxny::test_priority_queue2();
return 0;
}