优先级队列priority_queue
//需要建立小堆则需传入三个参数,第三个判断小堆(升序)
priority_queue<int,vector<int>,greater<int>> pq1;
pq1.push(1);
pq1.push(3);
pq1.push(9);
pq1.push(4);
pq1.push(7);
pq1.push(0);
pq1.push(2);
while(!pq1.empty())
{
cout << pq1.top() << " ";
pq1.pop();
}
cout << endl;
优先级队列就是以堆为基础的队列,当插入一个值时,自动建堆,默认为大堆less<>如果需要建立小堆,需要传入三个参数,默认第二个为vector类型,greater<>为建立小堆的仿函数。
每次top中的值总是极值,用堆来实现队列中数据的排序。
priority_queue的实现
class priority_queue
{
public:
//向上调整:尾部插入新值,找到合适位置
void adjust_up(int child)
{
int parent = (child - 1) / 2;
while (child > 0)
{
//传入两个要对比的值,在调用优先级队列时,传入需要建立的堆类型,然后调用仿函数,返回对应比较值
if (Compare()(_con[parent], _con[child]))//匿名函数
{
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())
{
if (child + 1 < _con.size()
&& Compare()(_con[child],_con[child + 1]))
{
++child;
}
if (Compare()(_con[parent], _con[child]))
{
swap(_con[child], _con[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
private:
Container _con;
}
优先级队列就是对堆的操作,所以向上和向下两个调整是必须的,每当执行pop和push时,数据变化都要调整堆:父节点大于子节点时交换,那么建立的就是小堆,父节点小于子节点时交换,那么建立的就是小堆,这是堆的调整,也是建堆的基础。
但是我们会发现,当我们在定义队列时,无法做到修改类中函数的大于小于号,所以可以建立仿函数,来让用户自定义是大堆还是小堆。
仿函数
//仿函数
template<class T>
struct less
{
bool operator()(const T& x, const T& y)
{
return x < y;
}
};
template<class T>
struct greater
{
bool operator()(const T& x, const T& y)
{
return x > y;
}
};
以前我们对堆调整时,需要判断数据的大小:
父节点大于子节点时交换,那么建立的就是小堆,父节点小于子节点时交换,那么建立的就是小堆,这里的大于小于号以前是需要手动变的,现在有了仿函数,只需要在实例化的时候传入greater或less参数,就能自动识别是大堆的建立还是小堆的建立。
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();
}
基本的操作实现。
针对类型构建仿函数:
void test2()
{
// 大堆,需要用户在自定义类型中提供<的重载
priority_queue<Date, vector<Date>, greater<Date>> q1;
q1.push(Date(2018, 10, 29));
q1.push(Date(2018, 10, 30));
q1.push(Date(2018, 10, 28));
cout << q1.top() << endl;
priority_queue<Date*, vector<Date*>, PDateGreater> q2;
q2.push(new Date(2018, 10, 29));//传的是地址,要新建仿函数
q2.push(new Date(2018, 10, 30));
q2.push(new Date(2018, 10, 28));
cout << *(q2.top()) << endl;
}
我们可以发现,q1传入的是Date类型的值,可以直接使用上面我们建立的仿函数;在q2中,当我们传入的是new出来的地址,那么就需要建立新的仿函数来传地址类型的,解引用后再比较大小。
class PDateLess
{
public:
bool operator()(const Date* p1, const Date* p2)
{
return *p1 < *p2;
}
};
class PDateGreater
{
public:
bool operator()(const Date* p1, const Date* p2)
{
return *p1 > *p2;
}
};