priority_queue 是优先级队列,又称堆,可以存储任意类型
模板参数 T 表示存储元素的类型,Container 表示底层使用的容器,Compare 表示存储元素的比较方法(less 建大堆,greater 建小堆)
一、priority_queue 类的模拟实现
priority_queue 类常用接口模拟实现:
//test.cpp
#include "priority_queue.h"
int main()
{
starrycat::priority_queue_test1();
return 0;
}
//priority_queue.h
#pragma once
#include <iostream>
#include <vector>
#include <assert.h>
using std::cout;
using std::endl;
namespace starrycat
{
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;
}
};
template<class T, class Container = std::vector<T>, class Compare = less<T>>
class priority_queue
{
public:
void AdjustUp(size_t child)
{
size_t parent = (child - 1) / 2;
while (child > 0)
{
if (Compare()(_con[parent], _con[child]))
{
std::swap(_con[child], _con[parent]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
void push(const T& x)
{
_con.push_back(x);
AdjustUp(_con.size() - 1);
}
void AdjustDown(size_t parent)
{
size_t child = parent * 2 + 1;
while (child < _con.size())
{
if (child + 1 < _con.size() && Compare()(_con[child], _con[child + 1]))
{
++child;
}
if (Compare()(_con[parent], _con[child]))
{
std::swap(_con[child], _con[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
void pop()
{
assert(!empty());
std::swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
AdjustDown(0);
}
const T& top() const
{
assert(!empty());
return _con[0];
}
size_t size() const
{
return _con.size();
}
bool empty() const
{
return _con.empty();
}
private:
Container _con;
};
void priority_queue_test1()
{
//priority_queue<int> pq;
priority_queue<int, std::vector<int>, greater<int>> pq;
pq.push(0);
pq.push(9);
pq.push(6);
pq.push(2);
pq.push(1);
pq.push(3);
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
}
}