c++中set容器的排序

set容器的排序

内置数据类型的排序规则

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

using namespace std;

//利用仿函数指定容器的排序规则
class MyCompare {
public:
    bool operator()(int v1, int v2) {
        return v1 > v2;
    }
};


//内置数据类型的排序规则
void test01() {

    set<int> s;
    s.insert(88);
    s.insert(-9);
    s.insert(1);
    s.insert(98);
    s.insert(909);

    //打印，默认是升序
    for (set<int>::iterator it = s.begin(); it != s.end(); it++) {//909 98 88 1 -9
        cout << *it << " ";
    }
    cout << endl;

    //指定排序规则为降序
    set<int,MyCompare> s2;
    s2.insert(88);
    s2.insert(-9);
    s2.insert(1);
    s2.insert(98);
    s2.insert(909);

    for (set<int,MyCompare>::iterator it = s2.begin(); it != s2.end(); it++) {//-9 1 88 98 909
        cout << *it << " ";
    }
    cout << endl;
}

int main() {
    test01();
    return 0;
}

自定义数据类型的排序规则

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

using namespace std;

class Person {
public:
    Person(string name, int age) {
        this->name = name;
        this->age = age;
    }

    string name;
    int age;
};

//自定义的数据类型，需要指定排序规则
class comparePerson {
public:
    bool operator()(const Person &p1, const Person &p2) {
        //按照年龄降序
        return p1.age > p2.age;
    }
};

void test01() {

    set<Person,comparePerson> s;

    //创建Person对象
    Person p1("刘备", 40);
    Person p2("关羽", 20);
    Person p3("张飞", 30);
    Person p4("诸葛亮", 18);

    //插入数据到set集合中
    s.insert(p1);
    s.insert(p2);
    s.insert(p3);
    s.insert(p4);

    for (set<Person,comparePerson>::iterator it = s.begin(); it != s.end(); it++) {
        cout << "姓名：" << it->name << "\t年龄：" << it->age << endl;
    }

}

int main() {
    test01();
    return 0;
}