STL-set/multiset容器

1. set/multiset容器基本概念

set和multiset容器实现是基于红黑树，红黑树是平衡二叉树的一种，所以Set的特性是，所有元素都会根据元素的键值自动被排序。Set的元素即是键值又是实值。Set不允许两个元素有相同的键值。

请问，我们可以通过set的迭代器改变set元素的值吗？不行，因为set元素值就是其键值，关系到set元素的排序规则。如果任意改变set元素值，会严重破坏set内部元素的组织。换句话说，set的iterator是一种const iterator.

set拥有和list某些相同的性质，当对容器中的元素进行插入操作或者删除操作的时候，操作之前所有的迭代器，在操作完成之后依然有效，被删除的那个元素的迭代器必然是一个例外。

multiset特性及用法和set完全相同，唯一的差别在于它允许键值重复。

2. set常用API

2.1 set构造函数

set<T> st;//set默认构造函数：
mulitset<T> mst; //multiset默认构造函数:
set(const set &st);//拷贝构造函数

2.2 set赋值操作

set& operator=(const set &st);//重载等号操作符
swap(st);//交换两个集合容器

2.3 set大小操作

size();//返回容器中元素的数目
empty();//判断容器是否为空

2.4 set插入和删除操作

insert(elem);//在容器中插入元素。
clear();//清除所有元素
erase(pos);//删除pos迭代器所指的元素，返回下一个元素的迭代器。
erase(beg, end);//删除区间[beg,end)的所有元素 ，返回下一个元素的迭代器。
erase(elem);//删除容器中值为elem的元素。

2.5 set查找操作

find(key);//查找键key是否存在,若存在，返回该键的元素的迭代器；若不存在，返回set.end();
count(key);//查找键key的元素个数
lower_bound(keyElem);//返回第一个key>=keyElem元素的迭代器。
upper_bound(keyElem);//返回第一个key>keyElem元素的迭代器。
equal_range(keyElem);//返回容器中lower_bound和upper_bound两个迭代器。

3. set的insert返回值

//插入操作返回值
void test01(){
 
    set<int> s;
    pair<set<int>::iterator,bool> ret = s.insert(10);
    if (ret.second){
        cout << "插入成功:" << *ret.first << endl;
    }
    else{
        cout << "插入失败:" << *ret.first << endl;
    }
    s.insert(20);
    ret = s.insert(10);
    if(ret.second){
        cout << "插入成功:" << *ret.first << endl;
    }
    else{
        cout << "插入失败:" << *ret.first << endl;
    }

}

set的insert函数返会一个pair<iterator, bool>类型的对组, 我们可以通过对组的第二个值，也就是second值来判断元素是否插入成果能够。上例中，insert第一次插入成功，第二次插入结果插入失败。

4. set排序规则

set容器默认对插入容器的元素从小到大排序，那么如果我们想从大到小怎么办呢？set容器是一个模板类，由第二个参数，我们可以传递规则给set容器来指定排序规则。见下面案例：

struct MyCompare02{
    bool operator()(int v1,int v2){
        return v1 > v2;
    }
};
 
//set从大到小
void test02(){
 
    srand((unsigned int)time(NULL));
    //我们发现set容器的第二个模板参数可以设置排序规则，默认规则是less<_Kty>
    set<int, MyCompare02> s;
    for (int i = 0; i < 10;i++){
        s.insert(rand() % 100);
    }
   
    for (set<int, MyCompare02>::iterator it = s.begin(); it != s.end(); it ++){
        cout << *it << " ";
    }
    cout << endl;
}

我们说set容器中的元素必须要被排序，如果我们存放的是自定义的数据类型，那么STL就无法知道如何排序了，那么程序在编译的时候必然出错，怎么做呢？和上例一样，我们自己写一个排序规则，升序或者降序比较规则，都可以。

//set容器中存放对象
class Person{
public:
    Person(string name,int age){
        this->mName = name;
        this->mAge = age;
    }
public:
    string mName;
    int mAge;
};
 
 
struct MyCompare03{
    bool operator()(const Person& p1,const Person& p2){
        return p1.mAge > p2.mAge;
    }
};
 
void test03(){
 
    set<Person, MyCompare03> s;
 
    Person p1("aaa", 20);
    Person p2("bbb", 30);
    Person p3("ccc", 40);
    Person p4("ddd", 50);
 
    s.insert(p1);
    s.insert(p2);
    s.insert(p3);
    s.insert(p4);
 
    for (set<Person, MyCompare03>::iterator it = s.begin(); it != s.end(); it++){
        cout << "Name:" << it->mName << " Age:" << it->mAge << endl;
    }
 
}