C++标准库扩展,给结构排序

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;

}








  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值