1、set是集合容器,其中的元素是唯一的,集合中元素按一定的顺序排序,元素的插入是按照排序规则进行插入的,所以不能指定插入位置;
2、set采用红黑树变体的数据结构实现,红黑树属于平衡二叉树,再插入与删除操作上比vector快
3、set不能采用[]或者at()的方式访问数据,也不能按[]的方式插入数据;
一、set的基本操作
void print(set<int>& s) {
set<int>::iterator it = s.begin();
while (it != s.end()) {
cout << *it << " ";
it++;
}
cout << endl;
}
void print(set<int, greater<int>>& s) {
set<int>::iterator it = s.begin();
while (it != s.end()) {
cout << *it << " ";
it++;
}
cout << endl;
}
void func1() {
set<int> s;
set<int, less<int>> s2;
set<int, greater<int>> s3;
for (int i = 0; i < 10; i++) {
int temp = rand();
s.insert(temp);
}
print(s);
for (int i = 0; i < 10; i++) {
int temp = rand();
s2.insert(temp);
}
print(s2);
for (int i = 0; i < 10; i++) {
int temp = rand();
s3.insert(temp);
}
print(s3);
}
输出结果:
set<int> s; //默认从小到大排序
set<int, less<int>> s2; //从小到大排序
set<int, greater<int>> s3; //从大到小排序
二、set的删除操作
set.clear(); //清空set中的所有元素
set.erase(pos); //删除pos位置的元素
set.erase(begin, end); //删除[begin, end)区间的元素
set.erase(elem); //删除值为elem的元素
void func1() {
set<int> s;
set<int, less<int>> s2;
set<int, greater<int>> s3;
for (int i = 0; i < 10; i++) {
int temp = rand();
s.insert(temp);
}
s.insert(10000);
s.insert(20000);
print(s);
s.erase(s.begin());
print(s);
set<int>::iterator it1 = s.begin();
it1++;
set<int>::iterator it2 = s.begin();
it2++;
it2++;
it2++;
it2++;
s.erase(it1, it2);
print(s);
s.erase(20000);
print(s);
print(s3);
}
输出结果:
三、自定义排序与insert的返回值
当set中的元素类型是自定义类型时,编译器不知道如何排序,这时候就要用仿函数(详细请看:C++之仿函数)编写一个自定义排序方式:
class Teacher {
public:
Teacher(string name, int age) :name(name), age(age) {}
public:
string name;
int age;
};
class cmpTeacher {
public:
bool operator()(const Teacher left, const Teacher right) {
return left.age < right.age;
}
};
void func2() {
Teacher t1("北丐", 50);
Teacher t2("西毒", 46);
Teacher t3("东邪", 41);
Teacher t4("南帝", 48);
set<Teacher, cmpTeacher> s;
s.insert(t1);
s.insert(t2);
s.insert(t3);
s.insert(t4);
set<Teacher>::iterator it = s.begin();
while (it != s.end()) {
cout << "姓名:" << it->name << " 年龄:" << it->age << endl;
it++;
}
}
输出结果:
可以看到这里实现了按照年龄排序的set集合,那么如果插入一个年龄相同,姓名不同的元素会怎么样呢,这个借助insert的返回值来查看:
void func3() {
Teacher t1("北丐", 50);
Teacher t2("西毒", 46);
Teacher t3("东邪", 41);
Teacher t4("南帝", 48);
Teacher t5("裘千仞", 46);
set<Teacher, cmpTeacher> s;
pair<set<Teacher, cmpTeacher>::iterator, bool> psb;
psb = s.insert(t1);
if (psb.second == true)
cout << "姓名: 北丐 年龄: 50 插入成功" << endl;
else
cout << "姓名: 北丐 年龄: 50 插入失败" << endl;
psb = s.insert(t2);
if (psb.second == true)
cout << "姓名: 西毒 年龄: 46 插入成功" << endl;
else
cout << "姓名: 西毒 年龄: 46 插入失败" << endl;
psb = s.insert(t3);
if (psb.second == true)
cout << "姓名: 东邪 年龄: 41 插入成功" << endl;
else
cout << "姓名: 东邪 年龄: 41 插入失败" << endl;
psb = s.insert(t4);
if (psb.second == true)
cout << "姓名: 南帝 年龄: 48 插入成功" << endl;
else
cout << "姓名: 南帝 年龄: 48 插入失败" << endl;
psb = s.insert(t5);
if (psb.second == true)
cout << "姓名: 裘千仞 年龄: 46 插入成功" << endl;
else
cout << "姓名: 裘千仞 年龄: 46 插入失败" << endl;
cout << "-----------------最终结果----------------------" << endl;
set<Teacher>::iterator it = s.begin();
while (it != s.end()) {
cout << "姓名:" << it->name << " 年龄:" << it->age << endl;
it++;
}
}
输出结果:
pair<set<Teacher, cmpTeacher>::iterator, bool> psb;
psb = s.insert(t1);
set的insert函数返回一个pair类型,pair类型的第一个参数是迭代器(这个迭代器类型要与set的迭代器类型一致),第二个参数是bool类型,表示插入成功与否,源码如下:
template<bool _Multi2 = _Multi,
enable_if_t<!_Multi2, int> = 0>
_Pairib insert(const value_type& _Val) //_Pairib是insert的返回类型
{ // try to insert node with value _Val, favoring right side
return (_Insert_nohint(false,
_Val, _Nil()));
}
typedef pair<iterator, bool> _Pairib; //_Pairib的定义
通过结果可以看到相同年龄的元素并没有插入到集合中去,因为我们自定义的排序方式是按照年龄来的,所以年龄必须是唯一的。
四、set的find,count,lower_bound,upper_bound,equal_range操作
void func4() {
set<int> s;
for (int i = 0; i < 10; i++)
s.insert(i + 1);
print(s);
set<int>::iterator it = s.find(3);
cout << "3的下标为:" << distance(s.begin(), it) << endl;
int num = s.count(3);
cout << "3出现的次数:" << num << endl;
set<int>::iterator it1 = s.lower_bound(5);
cout << "大于等于5的第一个元素:" << *it1 << endl;
set<int>::iterator it2 = s.upper_bound(5);
cout << "大于5的第一个元素:" << *it2 << endl;
pair<set<int>::iterator, set<int>::iterator> pii = s.equal_range(5);
cout << "大于等于5的第一个元素:" << *pii.first << endl;
cout << "大于5的第一个元素:" << *pii.second << endl;
}
set<int>::iterator it = s.find(3);
查找元素3的位置,并返回指向该位置的迭代器;
int num = s.count(3);
统计3出现的次数,因为set集合中的元素的唯一的,所以这里不可能大于1,出现则为1,没出现则为0;
set<int>::iterator it1 = s.lower_bound(5);
找到第一个大于等于元素5的元素,并返回指向这个元素的迭代器;
set<int>::iterator it2 = s.upper_bound(5);
找到第一个大于元素5的元素,并返回指向这个元素的迭代器;
pair<set<int>::iterator, set<int>::iterator> pii = s.equal_range(5);
equal_range会返回一个pair类型,这个pair类型第一个参数是等同于lower_bound返回的迭代器,第二次迭代器等同于upper_bound返回的迭代器,源码如下:
_Pairii equal_range(const key_type& _Keyval) //_Pairii 即为equal_range返回的类型
{ // find range equivalent to _Keyval in mutable tree
return (_Eqrange(_Keyval));
}
typedef pair<iterator, iterator> _Pairii; //_Pairii的定义
输出结果:
五、set的迭代器
set的迭代器是双向迭代器,只能做it++这样的单步操作,不能做it + 5这种操作。