用set管理自定义类型

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
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞