目录
template<class T, class Container = deque<T>>
class stack
{
public:
void push(const T& x)
{
_con.push_back(x);
}
void pop()
{
_con.pop_back();
}
const T& top()
{
return _con.back();
}
size_t size()
{
return _con.size();
}
bool empty()
{
return _con.empty();
}
private:
Container _con;
};
?
//适配器模式/配接器
template<class T, class Container = deque<T>>
class queue
{
public:
void push(const T& x)
{
_con.push_back(x);
}
void pop()
{
_con.pop_front();
}
const T& front()
{
return _con.front();
}
const T& back()
{
return _con.back();
}
size_t size()
{
return _con.size();
}
bool empty()
{
return _con.empty();
}
private:
Container _con;
};
函数声明
| 接口说明 |
priority_queue()?
|
构造一个空的优先级队列
|
empty() |
检测优先级队列是否为空,是返回
true
,否则返回
false
|
top() |
返回优先级队列中最大
(
最小元素
)
,即堆顶元素
|
push() |
在优先级队列中插入元素
x
|
pop() |
删除优先级队列中最大
(
最小
)
元素,即堆顶元素
|
#include <iostream>
#include <vector>
#include <functional>
#include <queue>
using namespace std;
int main()
{
vector<int> v{3, 7, 4, 2, 5, 6, 8, 9, 1, 0};
//默认大堆
priority_queue<int> pq;
//大于仿函数 - 小堆
//priority_queue<int, vector<int>, greater<int>> pq;
for (auto e : v)
{
pq.push(e);
}
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
return 0;
}
大堆输出:
小堆输出:
接口实现:
template<class T, class Container = vector<T>>
class priority_queue
{
public:
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;
};
堆的插入: (演示图表为小堆)
代码实现:向上调整
void Adjust_up(int child)
{
int parent = (child - 1) / 2;
while (child > 0)
{
if (_con[child] > _con[parent])
{
swap(_con[child], _con[parent]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
堆的删除:(演示图表为小堆)
代码实现:向下调整
void Adjust_down(int parent)
{
size_t child = parent * 2 + 1;
while (child < _con.size())
{
//child + 1 < _con.size() 是为了防止右孩子越界
if (child + 1 < _con.size() && _con[child + 1] > _con[child])
{
child++;
}
if (_con[child] > _con[parent])
{
swap(_con[child], _con[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
完整代码:
????????????????????????????????????????????????????????????????????????????????????????????????priority_queue.h? 初始版本
template<class T, class Container = vector<T>>
class priority_queue
{
public:
// 向上调整
void Adjust_up(int child)
{
int parent = (child - 1) / 2;
while (child > 0)
{
if (_con[child] > _con[parent])
{
swap(_con[child], _con[parent]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
void push(const T& x)
{
_con.push_back(x);
Adjust_up(_con.size() - 1);
}
//向下调整
void Adjust_down(int parent)
{
size_t child = parent * 2 + 1;
while (child < _con.size())
{
//child + 1 < _con.size() 是为了防止右孩子越界
if (child + 1 < _con.size() && _con[child + 1] > _con[child])
{
child++;
}
if (_con[child] > _con[parent])
{
swap(_con[child], _con[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
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;
};
//--------------------------------------------------------------
void test_priority_queue()
{
vector<int> v{3, 7, 4, 2, 5, 6, 8, 9, 1, 0};
//默认大堆
priority_queue<int> pq;
for (auto e : v)
{
pq.push(e);
}
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
}
运行结果:
? ? ? ? ?至此,完成priority_queue初始版本,但是这里存在一个问题,以上代码默认实现的是大堆,那如何修改为小堆呢?每次去修改向上/向下调整代码也能实现,但是太过繁琐,而在 stl 库里面,这里运用了一个语法:传入第三个模板参数---仿函数
进阶完整版代码如下:
????????????????????????????????????????????????????????????????????????????????????????????????????????priority_queue.h
#pragma once
namespace dwr
{
template<class T>
class less
{
public:
bool operator()(const T& x, const T& y)
{
return x < y;
}
};
template<class T>
class greater
{
public:
bool operator()(const T& x, const T& y)
{
return x > y;
}
};
template<class T, class Container = vector<T>, class Comapre = less<T>>
class priority_queue
{
public:
// 向上调整
void Adjust_up(int child)
{
Comapre com;
int parent = (child - 1) / 2;
while (child > 0)
{
//if (_con[child] > _con[parent])
if(com(_con[parent], _con[child])) // 注意: 默认是less 小于
{
swap(_con[child], _con[parent]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
void push(const T& x)
{
_con.push_back(x);
Adjust_up(_con.size() - 1);
}
//向下调整
void Adjust_down(int parent)
{
Comapre com;
size_t child = parent * 2 + 1;
while (child < _con.size())
{
//child + 1 < _con.size() 是为了防止右孩子越界
//if (child + 1 < _con.size() && _con[child + 1] > _con[child])
if (child + 1 < _con.size()
&& Comapre()( _con[child] , _con[child + 1])) //这里是匿名对象调用operator[]
{
child++;
}
if (com(_con[parent] , _con[child]))
{
swap(_con[child], _con[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
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;
};
//--------------------------------------------------------------
void test_priority_queue()
{
vector<int> v{3, 7, 4, 2, 5, 6, 8, 9, 1, 0};
//大于仿函数 - 小堆
priority_queue<int, vector<int>, greater<int>> pq;
for (auto e : v)
{
pq.push(e);
}
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
}
}
????????????????????????????????????????????????????????????????????????????????????????????????????????????????test.cpp
#include <iostream>
#include <vector>
#include <functional>
#include <queue>
using namespace std;
#include "priority_queue.h"
int main()
{
dwr::test_priority_queue();
return 0;
}
例:
????????1. stack和queue不需要遍历(因此stack和queue没有迭代器),只需要在固定的一端或者两端进行操作。
????????2. 在stack中元素增长时,deque比vector的效率高(扩容时不需要搬移大量数据);queue中的元素增长时,deque不仅效率高,而且内存使用率高。
以上仅代表个人观点,欢迎讨论