C++标准库的容器依赖less操作来比较元素的大小,比如std::sort可以传入一个pred,提供元素的less比较方法,缺省会使用std::less即调用元素的<操作符
对于自定义结构类型,可能需要根据一个或者多个数据成员来比较其大小,比如如下人员信息结构,需要结合三个字段进行排序,否则不能唯一确定大小:
struct Person
{
std::string strName; //名称
std::string strClass; //班级
std::string strDepart; //系
//构造函数
Person(const std::string& name, const std::string& clas, const std::string& dept);
};
为每个结构提供排序函数很繁琐,也容易出错,但可以使用模板来解决这一类问题,简单有效!
首先我们把比较函数定义成函数对象,并在内部保存参与比较的数据成员指针。对于使用单个数据成员来比较整个元素大小的函数对象,我们定义如下:
template<class TField>
struct SingleCompare
{
TField f_; //参与比较的数据成员指针
//比较函数
template<class T>
bool operator ()(const T& l, const T& r)
{
return l.*f_ < r.*f_;
}
//构造函数
SingleCompare(TField f) : f_(f){}
};
下面再定义一个函数对象,用于在已有的大小比较函数对象上再增加一个参与比较的数据成员:
template<class TField, class TBaseCompare>
struct AddCompareField
{
TField f_; //新增参与比较的数据成员指针
TBaseCompare b_; //基础比较对象
//比较函数
template<class T>
bool operator ()(const T& l, const T& r)
{
return (l.*f_ < r.*f_) ||((l.*f_ == r.*f_) && b_(l,r));
}
//构造函数
AddCompareField(TField f, TBaseCompare b) : f_(f),b_(b){}
};
借助上面的AddCompareField类,我们就可以方便的扩展出对多个数据成员进行比较的函数对象:
template<class F1, class F2>
struct Comp2 : AddCompareField<F1,SingleCompare<F2> >
{
Comp2(F1 f1, F2 f2):AddCompareField(f1, SingleCompare<F2>(f2)){}
};
template<class F1, class F2, class F3>
struct Comp3 : AddCompareField<F1,Comp2<F2,F3> >
{
Comp3(F1 f1, F2 f2, F3 f3):AddCompareField(f1, Comp2<F2,F3>(f2,f3)){}
};
。。。4个、5个比较成员的以此类推。
为了方便调用,继续定义一组比较对象的构造器函数:
template<class F1>
SingleCompare<F1> make_less_compare(F1 f1){
return SingleCompare<F1>(f1);
}
template<class F1, class F2>
Comp2<F1,F2> make_less_compare(F1 f1, F2 f2){
return Comp2<F1,F2>(f1, f2);
}
template<class F1, class F2, class F3>
Comp3<F1,F2,F3> make_less_compare(F1 f1, F2 f2, F3 f3){
return Comp3<F1,F2,F3>(f1, f2, f3);
}
这样我们就可以对包含Person的容器进行排序了,测试代码如下:
int main()
{
std::vector<Person> persons;
persons.push_back(Person("=xiaowang", "class3", "computer"));
persons.push_back(Person("=xiaoli", "class1", "math"));
persons.push_back(Person("=xiaozhu", "class2", "computer"));
//按班级排序
std::sort(persons.begin(), persons.end(), make_less_compare(&Person::strClass));
//按名称排序
std::sort(persons.begin(), persons.end(), make_less_compare(&Person::strName));
//按系,班级,名称排序
std::sort(persons.begin(), persons.end(), make_less_compare(&Person::strDepart, &Person::strClass, &Person::strName));
return 0;
}