最大堆和最小堆又叫大顶堆和小顶堆(大根堆和小根堆),是二叉树的一种排列方式。
最大堆:根结点的键值是所有堆结点键值中最大者,且每个结点的值都比其孩子的值大。
最小堆:根结点的键值是所有堆结点键值中最小者,且每个结点的值都比其孩子的值小。
堆排序的过程就是不断构造最大堆的过程。堆排序的时间复杂度是O(nlogn).
C++中的优先队列priority_queue可以用来构造最大堆和最小堆。在优先队列中,元素被赋予优先级,当访问优先队列时,优先级最高的优先被访问,即top()访问的优先级最高的元素。int元素默认越大优先级越高。
1 //升序队列,小顶堆
2 priority_queue <int, vector<int>, greater<int>> q;
3 //降序队列,大顶堆
4 priority_queue <int, vector<int>, less<int>>q;
5
6 //greater和less是std实现的两个仿函数(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了)
对于更复杂的数据结构,如结构体,需要自定义优先级排序。
struct node
{
string name;
int value;
node(string name_, int value_) : name(name_), value(value_) {}
};
struct compare
{
bool operator()(const node& a, const node& b)
{
return a.value < b.value; // 升序
}
};
priority_queue<node, vector<node>, compare> q{};