集合(set)实现了快速查找元素、插入元素以及删除元素。具体实现则采用了红黑树的二叉平衡搜素。
特点:
1.元素不重复。
2.采用中序遍历算法,使得效率比vector,map,list高。
3.自动按照key的大小排序。
一.创建set集合对象
在创建set对象时,与其他STL的容器一样,需要指定类型。
#include <iostream>
#include <set>
using namespace std;
set<int>s;
如果set的类型为自定义类型,那么在使用set之前,必须要重载运算符
<
<script type="math/tex" id="MathJax-Element-1"><</script>。在这里要尤其注意。
错误重载方法:
#include <set>
#include <cstdio>
using namespace std;
struct node
{
int a ,b ,c ;
node(){}
node(const int a1,const int b1,const int c1)
{a=a1 ,b=b1 ,c=c1 ;}
bool operator < (const node &node2)const
{
return a<node2.a;
}
};
set<node>s;
int main()
{
s.insert(node(1,2,3));
s.insert(node(1,3,4));
for(set<node>::iterator p=s.begin();p!=s.end();p++)
printf("%d %d %d\n",p->a ,p->b ,p->c );
return 0;
}
让我们看一下运行结果:
而不是
1
正确写法
#include <set>
#include <cstdio>
using namespace std;
struct node
{
int a ,b ,c ;
node(){}
node(const int a1,const int b1,const int c1)
{a=a1 ,b=b1 ,c=c1 ;}
bool operator < (const node &node2)const
{
if(a!=node2.a)return a<node2.a;
if(b!=node2.b)return b<node2.b;
return c<node2.c;
}
};
set<node>s;
int main()
{
s.insert(node(1,2,3));
s.insert(node(1,3,4));
for(set<node>::iterator p=s.begin();p!=s.end();p++)
printf("%d %d %d\n",p->a ,p->b ,p->c );
return 0;
}
所以说,不能偷懒少些代码。 O__O”…
二.插入
1.insert()
①insert(a)具体使用方法见上面的代码。
②insert(begin,end)将begin到end之间的元素插入set中
set<int>s;
int num[]={1,2,3};
int main()
{
s.insert(num,num+3);
return 0;
}
三.查找
1.find()
find(a) 返回元素a的地址
set<int>s;
int num[]={2,1,3};
int main()
{
s.insert(num,num+3);
printf("%d\n",*(s.find(1)));
return 0;
}
2.count()
count(a) 返回元素a的个数。然而蒟蒻认为该函数并没有什么用,因为在set中元素的个数不是1就是0,不过可以判断元素a是否存在。
set<int>s;
int num[]={2,1,3};
int main()
{
s.insert(num,num+3);
printf("%d\n",s.count(1));
return 0;
}
四.删除 erase()函数
erase(a) 删除键为a的元素
erase(begin,end) 删除begin到end之间的元素
erase(iterator) 删除迭代器所指向的元素
#include <iostream>
#include <cstdio>
#include <set>
using namespace std;
set<int>s;
int num[]={2,1,3,4,6,10};
int main()
{
s.insert(num,num+6);
//第一种方式
s.erase(1);
for(set<int>::iterator p=s.begin();p!=s.end();++p)
printf("%d ",*p);
puts("");
//第二种方式
set<int>::iterator Begin=s.begin() ,End=s.end();
++Begin ,--End ;
s.erase(Begin,End);
for(set<int>::iterator p=s.begin();p!=s.end();++p)
printf("%d ",*p);
puts("");
//第三种方式
s.erase(s.find(2));
for(set<int>::iterator p=s.begin();p!=s.end();++p)
printf("%d ",*p);
puts("");
return 0;
}
五.常用函数
begin() rend() 返回set的初始地址
end() rbegin() 返回set的结束地址
clear() 清空set
size() 返回set中元素个数
empty() 判断set是否为空。若为空,返回true
六.遍历
使用迭代器iterator
for(set<int>::iterator p=s.begin();p!=s.end();++p)