C ++在容器类中使用自定义比较器进行排序

时间:2017-11-09 18:48:15

标签: c++ sorting set comparison

我有一个名字和年龄的人。

我有一个名为people的容器类,用于存储一组人员。

我创建了两个自定义比较器,按名称和年龄排序。

当我在我的类(People)中存储包含Persons的集合时,如何传递自定义比较器。

例如

我的比较器看起来像这样

struct compareByName
{
    bool operator()(const Person & Left, const Person & Right)
    {
        return (Left.getName() < Right.getName());
    }
};

在主要内容如果我想按姓名对一组人进行排序我只是做

set<Person, compareByName> peopleByName;

或我的年龄

set<Person, compareByAge> peopleByAge;

我遇到问题的地方是如何在我的人员容器类中使用它 我会有像

这样的东西
class People
{
private:
    set<Person, COMPARATOR> m_people;
}

COMPARATOR可以是名字还是年龄

2 个答案:

答案 0 :(得分:4)

您可以使用std::function作为比较器类型,然后为构造函数提供特定的比较器:

class People
{
    using PeopleSet = set<Person, std::function<bool(const Person &p1, const Person &p2 )>>;

    People() : people( compareByName() ) {}
    void sortByAge();
private:
    PeopleSet people; 
};

请注意,创建set后无法更改比较器,您必须创建另一个实例:

void People::sortByAge()
{
    people = PeopleSet( people.begin(), people.end(), compareByAge() );
}

这将涉及复制或移动整套。如果您希望能够同时使用两种方式,请改用boost::multi_index

答案 1 :(得分:1)

您可以通过模板进行操作:

#include <iostream>
#include <string>
#include <set>

class Person 
{
private:
    std::string name;
    int age;
public:
    Person(std::string name, int age) : name(name), age(age) {};
    std::string getName() const { return name; };
    int getAge() const { return age; };
};

template<class COMPARATOR> class People
{
private:
    std::set<Person, COMPARATOR> m_people;
public:
    void AddPerson(Person const &p)
    {
        m_people.insert(p);
    }
    void PrintPeople()
    {
        for (Person p : m_people)
        {
            std::cout << p.getName() << " - " << p.getAge() << std::endl;
        }
    }
};

struct SortedByName
{
    bool operator()(const Person & Left, const Person & Right)
    {
        return (Left.getName() < Right.getName());
    }
};

struct SortedByAge
{
    bool operator()(const Person & Left, const Person & Right)
    {
        return (Left.getAge() < Right.getAge());
    }
};


int main()
{
    Person p1("Bob", 21);
    Person p2("Jack", 13);
    Person p3("Kath", 33);

    People<SortedByName> ppl1;
    ppl1.AddPerson(p1);
    ppl1.AddPerson(p2);
    ppl1.AddPerson(p3);
    std::cout << "By Name:" << std::endl;
    ppl1.PrintPeople();

    std::cout << std::endl;

    People<SortedByAge> ppl2;
    ppl2.AddPerson(p1);
    ppl2.AddPerson(p2);
    ppl2.AddPerson(p3);
    std::cout << "By Age:" << std::endl;
    ppl2.PrintPeople();

    return 0;
}

打印:

By Name:
Bob - 21
Jack - 13
Kath - 33

By Age:
Jack - 13
Bob - 21
Kath - 33