很多STL算法都使用了function objects也叫函数对象functors,是一个对象,重载了()操作符,也就能像调用函数那样了使用这个对象了。
比如简单的sort算法:
vector<int> vec;
......
sort(vec.begin(), vec.end());
那就可以排序vec,但是如果vec装的不是int而是自定义类型呢。
如LeetCode上最喜欢用的链表数据结构:
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(nullptr) {}
};
如果我们需要比较ListNode的大小,那么就可以用三种方法实现:
方法1 定义一个adaptor功能的数据结构
struct adaptor
{
ListNode *a;
bool operator<(adaptor n)
{
return a->val < n.a->val;
}
};
那就可以排序了:
void sortVec(vector<adaptor> &vln)
{
sort(vln.begin(), vln.end());
}
方法2 我们也可以定义一个比较函数来实现这个功能:
bool compList(const ListNode t1, const ListNode t2)
{
return t1.val < t2.val;
}
sort(vl.begin(), vl.end(), compList);
方法3 定义一个functor来实现:
struct functor
{
bool operator()(ListNode a, ListNode b)
{
return a.val < b.val;
}
};
sort(vl.begin(), vl.end(), functor());
有时候也许我们需要把一个双参数的操作函数改为单个参数的函数,那么也可以定义一个类来实现,如:
template <class T>
bool tooBig(const T & val, const T & lim)
{
return val > lim;
}
template<class T>
class TooBig2
{
private:
T cutoff;
public:
TooBig2(const T & t) : cutoff(t) {}
bool operator()(const T & v) { return tooBig<T>(v, cutoff); }
};
这样双值比较就可以如下实现了:
TooBig2<int> tB100(100);
int x;
cin >> x;
if (tB100(x)) // same as if (tooBig(x,100))