《priority_queue的模拟实现》
本文主要介绍
文章目录
- 一、仿函数
-
- 1.1 仿函数的定义
- 1.2 普通仿函数
- 1.3 需要自己实现仿函数
- 二、priority_queue的模拟实现
一、仿函数
1.1 仿函数的定义
所谓的仿函数(functor)，是通过重载()运算符模拟函数形为的类。
因此，这里需要明确两点：
- 仿函数不是函数，它是个类；
- 仿函数重载了()运算符，使得它的对你可以像函数那样子调用(代码的形式好像是在调用函数)。（C语言中就是函数指针）
1.2 普通仿函数
struct Less
{
bool operator()(int x,int y)
{
return x < y;
}
};
struct Greater
{
bool operator()(int x,int y)
{
return x > y;
}
};
int main()
{
// Less就叫仿函数类型 less就叫函数对象
Less less;
cout << less(1, 2) << endl;
Greater gt;
cout << gt(1, 2) << endl;
cout << gt.operator()(1,2) << endl;//两者等价
return 0;
}
为了适应更多的类型可以加入模板
//仿函数
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;
}
};
int main()
{
// Less就叫仿函数类型 less就叫函数对象
Less<int> less;
cout << less(1, 2) << endl;
Greater<int> gt;
cout << gt(1, 2) << endl;
cout << gt.operator()(1,2) << endl;
cout << Greater<double>()(3.14, 8.9) << endl;// 可以用匿名对象
return 0;
}
1.3 需要自己实现仿函数
如果在priority_queue中放自定义类型的数据，用户需要在自定义类型中提供<或者>的重载
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);
private:
int _year;
int _month;
int _day;
};
ostream& operator<<(ostream& _cout, const Date& d)
{
_cout << d._year << "-" << d._month << "-" << d._day;
return _cout;
}
void test_priority_queue2()
{
//priority_queue pq;
priority_queue<Date,vector<Date>,greater<Date>> pq;
pq.push(Date(2022, 1, 1));
pq.push(Date(2022, 1, 2));
pq.push(Date(2022, 1, 3));
pq.push(Date(2022, 1, 4));
while (!pq.empty())
{
cout << pq.top() << " ";
pq.pop();
}
cout << endl;
}
此处我们想存一个日期类进行比较，我们如果使用默认的less或者greater，他就会去比较两个日期类对象，但是我们想要比较的是具体的日期大小。
所以我们自定义的类型中要提供< 或者 >的重载，在进行比较时，会去调用用户提供的<或者>的重载函数
还有什么特殊情况下，需要我们自己实现仿函数吗？
如下例子：我们在priority_queue中存放自定义类型的指针，我们需要比较的是指针就引用以后的具体日期。
void test_priority_queue2()
{
priority_queue<Date*> pq;
//priority_queue,greater> pq;
pq.push(new Date(2022, 1, 9));
pq.push(new Date(2022, 1, 2));
pq.push(new Date(2022, 1, 3));
pq.push(new Date(2022, 1, 4));
while (!pq.empty())
{
cout << *pq.top() << " ";
pq.pop();
}
cout << endl;
}
此时的less或者greater直接取的Date里面的地址进行比较，由于new地址是堆上的空闲随机地址，故这是不可行的，需要自己实现仿函数
class Date
{
friend class MyLess;//类外拿不到成员变量，搞成友元的
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);
private:
int _year;
int _month;
int _day;
};
ostream& operator<<(ostream& _cout, const Date& d)
{
_cout << d._year << "-" << d._month << "-" << d._day;
return _cout;
}
struct MyLess
{
bool operator()(const Date* d1, const Date* d2)const
{
return (d1->_year < d2->_year) ||
(d1->_year == d2->_year && d1->_month < d2->_month) ||
(d1->_year == d2->_year && d1->_month == d2->_month && d1->_day < d2->_day);
}
};
void test_priority_queue2()
{
priority_queue<Date*,vector<Date*>,MyLess> pq;
//priority_queue,greater> pq;
pq.push(new Date(2022, 1, 9));
pq.push(new Date(2022, 1, 2));
pq.push(new Date(2022, 1, 3));
pq.push(new Date(2022, 1, 4));
while (!pq.empty())
{
cout << *pq.top() << " ";
pq.pop();
}
cout << endl;
}
总结：
如果数据类型，不支持比较，或者比较的方式，不是你想要的那么可以自己实现仿函数，按照自己想要的方式去比较，控制比较逻辑
二、priority_queue的模拟实现
priority_queue底层是堆的结构，因此只需要对堆进行封装即可
#pragma once
#include