用set管理自定义类型

2019独角兽企业重金招聘Python工程师标准>>>

set内部使用红黑树保存对象，而红黑树的构建需要对元素进行排序，所以在向set容器插入数据时，需要保证插入的数据是可排序的，或者提供排序方法供set调用来对数据排序

class KeyValue
{
public:
    int key;
    int value;
};

上面是一个简单的自定义类型，如果要使用set来管理KeyValue的对象集合，需要如下操作

// 类型名可以任意
class Less
{
public:
    bool operator() (KeyValue &a, KeyValue &b)
    {
        return (a.key < b.key);
    }
};


set<KeyValue, Less> obj;

这样在插入数据时，set会调用 Less对象的 operator() 函数对插入数据进行排序

为了维持树结构，保证其不会被破坏，所以set中的元素是无法修改的，只能 erase 后再 insert

转载于:https://my.oschina.net/tigerBin/blog/1186763