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;