c++ set 常用用法介绍

set 集合容器：实现了红黑树的平衡二叉检索树的数据结构，插入元素时，它会自动调整二叉树的排列，把元素放到适当的位置，以保证每个子数根节点键值大于左子树所有节点的键值，小于右子树所有节点的键值；另外，还得保证根节点左子树的高度与右子树高度相等。

平衡二叉检索树使用中序遍历算法，检索效率高于 vector、deque 和 list 等容器，另外使用了中序遍历可将键值按照从小到大遍历出来。

构造 set 集合主要目的是为了快速检索，不可直接去修改键值。

常用操作：

元素插入：insert()

中序遍历：类似 vector 遍历（用迭代器）

反向遍历：利用反向迭代器 reverse_iterator, 例如：

set<int> s;
......
set<int>::reverse_iterator rit;
for(rit=s.rbegin();rit!=s.rend();rit++)

元素删除：与插入一样，可以高效的删除，并自动调整使红黑树平衡。

set<int> s;
s.erase(2);   // 删除键值为2的元素
s.clear();

元素检索：find()，若找到，返回该键值迭代器的位置，否则，返回最后一个元素后面一个位置。

set<int>::iterator it;
it=s.find(5);  // 查找键值为5的元素
if(it!=s.end())  // 找到
cout<<*it << endl;
else                // 未找到
cout<<"未找到"；

自定义比较函数

(1) 元素不是结构体：

例：

// 自定义比较函数 myComp, 重载“()”操作符

struct myComp{
bool operator()(const your_type &a, const your_type &b){
return a.data-b.data>0;
}
}
set< int,myComp >s;
.....
set < int,myComp > :: iterator it;

(2)如果元素是结构体，可以直接将比较函数写在结构体内。

例：

struct Info
{
string name;
float score;
// 重载"<"操作符，自定义排序规则
bool operator<(const Info &a) const{
//  按score 从大到小排列
return a.score < score;
}
}
set< Info >s;
......
set<Info>::iterator it;