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