🏖️作者:@malloc不出对象
⛺专栏:C++的学习之路
👦个人简介:一名双非本科院校大二在读的科班编程菜鸟,努力编程只为赶上各位大佬的步伐🙈🙈
目录
- 前言
- 一、priority_queue的介绍
- 二、priority_queue的使用
- 三、仿函数
- 四、priority_queue的模拟实现
前言
本篇文章讲解的是优先级队列的使用以及模拟实现。
一、priority_queue的介绍
- 优先队列是一种容器适配器,根据严格的弱排序标准,它的第一个元素总是它所包含的元素中最大的。
- 此上下文类似于堆,在堆中可以随时插入元素,并且只能检索最大堆元素(优先队列中位于顶部的元素)。
- 优先队列被实现为容器适配器,容器适配器即将特定容器类封装作为其底层容器类,queue提供一组特定的成员函数来访问其元素。元素从特定容器的“尾部”弹出,其称为优先队列的顶部。
- 底层容器可以是任何标准容器类模板,也可以是其他特定设计的容器类。容器应该可以通过随机访问迭代器访问,并支持以下操作:
empty():检测容器是否为空
size():返回容器中有效元素个数
front():返回容器中第一个元素的引用
push_back():在容器尾部插入元素
pop_back():删除容器尾部元素- 标准容器类vector和deque满足这些需求。默认情况下,如果没有为特定的priority_queue类实例化指定容器类,则使用vector。
- 需要支持随机访问迭代器,以便始终在内部保持堆结构。容器适配器通过在需要时自动调用算法函数make_heap、push_heap和pop_heap来自动完成此操作。
二、priority_queue的使用
优先级队列默认使用vector作为其底层存储数据的容器,在vector上又使用了堆算法将vector中元素构造成堆的结构,因此priority_queue就是堆,所有需要用到堆的位置,都可以考虑使用priority_queue。注意:默认情况下priority_queue是大堆。
函数声明 | 接口说明 |
---|---|
empty() | 检测优先级队列是否为空,是返回 true,否则返回 false |
top() | 返回优先级队列中最大元素(最小元素),即堆顶元素 |
push(x) | 在优先级队列中插入元素 x |
pop() | 删除优先级队列中最大元素(最小元素),即堆顶元素 |
size() | 返回优先级队列中元素的个数 |
下面我们来简单使用一下priority_queue:
#include <iostream>
#include <queue>
#include <vector>
#include <functional>
using namespace std;
void test()
{
// 默认是大堆, 底层是按照小于来进行比较的
priority_queue<int, vector<int>> pq1;
pq1.push(1);
pq1.push(3);
pq1.push(0);
pq1.push(7);
pq1.push(2);
while (!pq1.empty())
{
cout << pq1.top() << " ";
pq1.pop();
}
cout << endl;
// 要想创建小堆,此时我们应该在三个模板参数显式传递greater仿函数,它的底层是按照大于来进行比较的,我们需要包含functional这个头文件才能使用
priority_queue<int, vector<int>, greater<int>> pq2;
pq2.push(1);
pq2.push(3);
pq2.push(0);
pq2.push(7);
pq2.push(2);
while (!pq2.empty())
{
cout << pq2.top() << " ";
pq2.pop();
}
cout << endl;
}
int main()
{
test();
return 0;
}
如果在priority_queue中放自定义类型的数据,用户需要在自定义类型中提供> 或者< 的重载。
#include <iostream>
#include <queue>
#include <functional>
using namespace std;
class Date
{
public:
Date(int year = 1900, int month = 1, int day = 1)
: _year(year)
, _month(month)
, _day(day)
{}
bool operator<(const Date& d)const
{
return (_year < d._year) ||
(_year == d._year && _month < d._month) ||
(_year == d._year && _month == d._month && _day < d._day);
}
bool operator>(const Date& d)const
{
return (_year > d._year) ||
(_year == d._year && _month > d._month) ||
(_year == d._year && _month == d._month && _day > d._day);
}
friend ostream& operator<<(ostream& _cout, const Date& d)
{
_cout << d._year << "-" << d._month << "-" << d._day;
return _cout;
}
private:
int _year;
int _month;
int _day;
};
void TestPriorityQueue()
{
// 大堆,需要用户在自定义类型中提供<的重载
priority_queue<Date> q1;
q1.push(Date(2018, 10, 29));
q1.push(Date(2018, 10, 28));
q1.push(Date(2018, 10, 30));
while (!q1.empty())
{
cout << q1.top() << " ";
q1.pop();
}
cout << endl;
// 如果要创建小堆,需要用户提供>的重载
priority_queue<Date, vector<Date>, greater<Date>> q2;
q2.push(Date(2018, 10, 29));
q2.push(Date(2018, 10, 28));
q2.push(Date(2018, 10, 30));
while (!q2.empty())
{
cout << q2.top() << " ";
q2.pop();
}
cout << endl;
}
int main()
{
TestPriorityQueue();
return 0;
}
优先级队列的使用成本很低,下面我们来做一道题吧:
LeetCode 215. 数组中的第K个最大元素
这题使用优先级队列可谓是非常的简单,题目要求第K大的数,我们直接利用优先级队列建立一个大堆,再pop掉前K-1个数,此时栈顶元素就是最大值也是第K大的数了。
首先建堆的时间复杂度为O(N),然后调整建堆的时间复杂度为O(logN)循环K次,所以最终这种解决方案的时间复杂度为O(N + K * logN),如果N很大时需要很大的空间,那么还没有更优的解法呢?
我们可以考虑只建一个K大小的小堆这样时间复杂度就为O(K)了,比起O(N)来说可以节省不少的空间,建一个K大小的小堆,遍历后N - K个元素,如果它大于栈顶元素就加入进来调整建堆,最后你会发现前K个大的数都在这个小堆中,而栈顶元素就为这K个元素当中最小的那个,也就是第K大的元素!!总的时间复杂度为O(K + (N - K) * logK).
三、仿函数
仿函数(Functor)是一种可以像函数一样被调用的对象,它是一个类或者结构体,它实现了函数调用运算符(operator()),可以像普通函数一样被调用。与函数不同,仿函数可以存储状态并且可以在多次调用之间保持其状态。
此外,仿函数可以通过模板参数进行参数化,以支持不同类型的参数和返回类型,使其更加灵活。
在C++中,仿函数通常被用于泛型编程中的算法函数中,这些算法函数可以接受仿函数作为参数,从而实现不同的算法行为。通过使用仿函数,我们可以在运行时动态地决定算法的行为,这种灵活性使得C++中的泛型编程更加强大。
#include <iostream>
using namespace std;
template<class T>
struct Less
{
bool operator()(const T& x, const T& y)
{
return x < y;
}
};
template<class T>
class Greater
{
public:
template<class T>
bool operator()(const T& x, const T& y)
{
return x > y;
}
};
int main()
{
Less<int> lessFunc; // LessFunc对象就是一个仿函数,它可以像函数一样被调用
cout << lessFunc(7, 2) << endl;
cout << lessFunc.operator()(7, 2) << endl;
cout << Less<int>()(2, 7) << endl; // 匿名对象调用operator()函数
Greater<double> greaterFunc;
cout << greaterFunc(2.0, 1.2) << endl;
cout << greaterFunc.operator()(3.4, 4.6) << endl;
cout << Greater<double>()(2.0, 1.2) << endl;
return 0;
}
仿函数可以是类对象也可以是结构体对象,它也经常与我们的函数指针进行对比,我们的函数指针常用于回调函数之中,它并不是直接去调用那个函数,而是通过在一个函数中通过函数指针去间接那个函数,比起仿函数它还需要写函数参数以及返回值类型,这一点可能会给我们带来极大的困难,而我们的仿函数此时就显得非常好用了。
下面我们来看看这段代码,我们想实现的是日期类的优先级队列:
#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <functional>
using namespace std;
class Date
{
public:
Date(int year = 1900, int month = 1, int day = 1)
: _year(year)
, _month(month)
, _day(day)
{}
bool operator<(const Date& d)const
{
return (_year < d._year) ||
(_year == d._year && _month < d._month) ||
(_year == d._year && _month == d._month && _day < d._day);
}
bool operator>(const Date& d)const
{
return (_year > d._year) ||
(_year == d._year && _month > d._month) ||
(_year == d._year && _month == d._month && _day > d._day);
}
friend ostream& operator<<(ostream& _cout, const Date& d)
{
_cout << d._year << "-" << d._month << "-" << d._day;
return _cout;
}
private:
int _year;
int _month;
int _day;
};
void TestPriorityQueue()
{
// 大堆
priority_queue<Date*> q1;
q1.push(new Date(2018, 10, 29));
q1.push(new Date(2018, 10, 28));
q1.push(new Date(2018, 10, 30));
while (!q1.empty())
{
cout << *(q1.top()) << endl;
q1.pop();
}
cout << endl;
// 小堆
priority_queue<Date*, vector<Date*>, greater<Date*>> q2;
q2.push(new Date(2018, 10, 29));
q2.push(new Date(2018, 10, 28));
q2.push(new Date(2018, 10, 30));
while (!q2.empty())
{
cout << *(q2.top()) << endl;
q2.pop();
}
cout << endl;
}
int main()
{
TestPriorityQueue();
return 0;
}
我们来看看结果:
从上图我运行了三次,三次的结果都不同??这是为何??
这是因为库提供的仿函数不符合我们的要求,此时我们要比较的是日期类的值而非指针,就是我们比较的是指针,所以每次得出的结果都是不确定的,那么既然库中提供的仿函数不满足我们的需求,那我们就可以自行去实现一个。
#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <functional>
using namespace std;
class Date
{
public:
Date(int year = 1900, int month = 1, int day = 1)
: _year(year)
, _month(month)
, _day(day)
{}
bool operator<(const Date& d)const
{
return (_year < d._year) ||
(_year == d._year && _month < d._month) ||
(_year == d._year && _month == d._month && _day < d._day);
}
bool operator>(const Date& d)const
{
return (_year > d._year) ||
(_year == d._year && _month > d._month) ||
(_year == d._year && _month == d._month && _day > d._day);
}
friend ostream& operator<<(ostream& _cout, const Date& d)
{
_cout << d._year << "-" << d._month << "-" << d._day;
return _cout;
}
private:
int _year;
int _month;
int _day;
};
struct PDateLess
{
bool operator()(const Date* p1, const Date* p2)
{
return *p1 < *p2; // 比较日期
}
};
struct PDateGreater
{
bool operator()(const Date* p1, const Date* p2)
{
return *p1 > *p2;
}
};
void TestPriorityQueue()
{
// 大堆
priority_queue<Date*, vector<Date*>, PDateLess> q1;
q1.push(new Date(2018, 10, 29));
q1.push(new Date(2018, 10, 28));
q1.push(new Date(2018, 10, 30));
while (!q1.empty())
{
cout << *(q1.top()) << endl;
q1.pop();
}
cout << endl;
// 小堆
priority_queue<Date*, vector<Date*>, PDateGreater> q2;
q2.push(new Date(2018, 10, 29));
q2.push(new Date(2018, 10, 28));
q2.push(new Date(2018, 10, 30));
while (!q2.empty())
{
cout << *(q2.top()) << endl;
q2.pop();
}
cout << endl;
}
int main()
{
TestPriorityQueue();
return 0;
}
仿函数其实是C++中设计的非常好的一点,关于仿函数现阶段我们先就讲到这里,后续遇到了我们再来详谈。
四、priority_queue的模拟实现
// priority_queue.h
namespace curry
{
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;
}
};
template<class T, class Container = vector<int>, class Compare = less<T>>
class priority_queue
{
public:
priority_queue()
{}
template <class InputIterator> // 任意类型迭代器构造
priority_queue(InputIterator first, InputIterator last)
: _con(first, last)
{
int size = _con.size() - 1;
for (int i = (size - 1) / 2; i >= 0; --i) // 注意: 这里一定要使用int!如果使用size_t的话假设只有一个元素,--i就变为了-1,如果是size_t的话,i将会是一个很大的数此时造成死循环
ajustDown(i); //向下调整建堆
}
void ajustUp(int child)
{
Compare com;
int parent = (child - 1) / 2;
while (child > 0)
{
//if (_con[parent] < _con[child])
if (com(_con[parent], _con[child]))
{
swap(_con[parent], _con[child]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
void ajustDown(int parent)
{
Compare com;
size_t child = parent * 2 + 1;
while (child < _con.size())
{
//if (child + 1 < _con.size() && _con[child] < _con[child + 1])
if (child + 1 < _con.size() && com(_con[child], _con[child + 1]))
{
child++;
}
// if (_con[parent] < _con[child])
if (com(_con[parent], _con[child]))
{
swap(_con[parent], _con[child]);
parent = child;
child = 2 * parent + 1;
}
else
{
break;
}
}
}
void push(int x)
{
_con.push_back(x);
ajustUp(_con.size() - 1);
}
void pop()
{
swap(_con[0], _con[_con.size() - 1]);
_con.pop_back();
ajustDown(0);
}
const T& top()
{
return _con[0];
}
size_t size()
{
return _con.size();
}
bool empty()
{
return _con.empty();
}
private:
Container _con;
};
}
本篇文章的讲解就到这里了,如果有任何错处或者疑问欢迎大家评论区交流哦~~ 🙈 🙈