STL中set用法

头文件

#include  <set>

介绍

每个元素最多只出现一次，并且set中的元素已经从小到大排好序。
所有元素都会在插入时自动被排序

常用的函数

• begin() 返回set容器的第一个元素的地址
• end() 返回set容器的最后一个元素的地址
• clear() 删除set容器中所有的元素
• empty() 判断set容器是否为空
• max_size() 返回set容器可能包含的元素最大个数
• size() 返回当前set容器中的元素个数
• erase(it) 删除迭代器指针it处元素
• count() 用来查找set中某个元素出现的次数，因为一个键值在set只可能出现0或者1次
• find() 用来查找set中某个元素出现的位置，如果找到就返回这个元素的迭代器，如果这个元素不存在，则返回s.end()(最后一个元素的下一个位置)。

常用的介绍

set构造跟赋值

• set<T> st; //默认构造函数：
• set(const set &st); //拷贝构造函数
• set& operator=(const set &st); //重载等号操作符

#include <set>

void printSet(set<int> & s)
{
	for (set<int>::iterator it = s.begin(); it != s.end(); it++)
	{
		cout << *it << " ";
	}
	cout << endl;
}

//构造和赋值
void test01()
{
	set<int> s1;

	s1.insert(10);
	s1.insert(30);
	s1.insert(20);
	s1.insert(40);
	printSet(s1);

	//拷贝构造
	set<int>s2(s1);
	printSet(s2);

	//赋值
	set<int>s3;
	s3 = s2;
	printSet(s3);
}

int main() {

	test01();

	system("pause");

	return 0;
}

总结：

• set容器插入数据时用insert
• set容器插入数据的数据会自动排序

set大小与交换

• size(); //返回容器中元素的数目
• empty(); //判断容器是否为空
• swap(st); //交换两个集合容器

#include <set>

void printSet(set<int> & s)
{
	for (set<int>::iterator it = s.begin(); it != s.end(); it++)
	{
		cout << *it << " ";
	}
	cout << endl;
}

//大小
void test01()
{

	set<int> s1;
	
	s1.insert(10);
	s1.insert(30);
	s1.insert(20);
	s1.insert(40);

	if (s1.empty())
	{
		cout << "s1为空" << endl;
	}
	else
	{
		cout << "s1不为空" << endl;
		cout << "s1的大小为： " << s1.size() << endl;
	}

}

//交换
void test02()
{
	set<int> s1;

	s1.insert(10);
	s1.insert(30);
	s1.insert(20);
	s1.insert(40);

	set<int> s2;

	s2.insert(100);
	s2.insert(300);
	s2.insert(200);
	s2.insert(400);

	cout << "交换前" << endl;
	printSet(s1);
	printSet(s2);
	cout << endl;

	cout << "交换后" << endl;
	s1.swap(s2);
	printSet(s1);
	printSet(s2);
}

int main() {

	//test01();

	test02();

	system("pause");

	return 0;
}

总结：

• 统计大小 — size
• 判断是否为空 — empty
• 交换容器 — swap

set插入与删除

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

#include <set>

void printSet(set<int> & s)
{
	for (set<int>::iterator it = s.begin(); it != s.end(); it++)
	{
		cout << *it << " ";
	}
	cout << endl;
}

//插入和删除
void test01()
{
	set<int> s1;
	//插入
	s1.insert(10);
	s1.insert(30);
	s1.insert(20);
	s1.insert(40);
	printSet(s1);

	//删除
	s1.erase(s1.begin());
	printSet(s1);

	s1.erase(30);
	printSet(s1);

	//清空
	//s1.erase(s1.begin(), s1.end());
	s1.clear();
	printSet(s1);
}

int main() {

	test01();

	system("pause");

	return 0;
}

总结：

• 插入 — insert
• 删除 — erase
• 清空 — clear

set查找与统计

• find(key); //查找key是否存在,若存在，返回该键的元素的迭代器；若不存在，返回set.end();
• count(key); //统计key的元素个数

#include <set>

//查找和统计
void test01()
{
	set<int> s1;
	//插入
	s1.insert(10);
	s1.insert(30);
	s1.insert(20);
	s1.insert(40);
	
	//查找
	set<int>::iterator pos = s1.find(30);

	if (pos != s1.end())
	{
		cout << "找到了元素 ： " << *pos << endl;
	}
	else
	{
		cout << "未找到元素" << endl;
	}

	//统计
	int num = s1.count(30);
	cout << "num = " << num << endl;
}

int main() {

	test01();

	system("pause");

	return 0;
}

总结：

• 查找 — find （返回的是迭代器）
• 统计 — count （对于set，结果为0或者1）

set的遍历

  #include <iostream>
 2 #include<set>
 3 using namespace std;
 4 
 5 
 6 int main()
 7 {
 8     set<int> s;                //创建一个int类型的set
 9  
10     s.insert(10);                //插入数据
11     s.insert(30);
12     s.insert(20);
13     s.insert(40);                
14 
15     //遍历数据，用迭代器遍历数据
16     for (set<int>::iterator it = s.begin(); it != s.end(); ++it)    
17     {
18         cout << *it << endl;
19     }
20     //这里用到了set中的元素已经从小到大排好序的性质
21     
22     return 0;
23 }

重点

结构体类型struct的set，使用时必须重载‘<’运算符

 1 #include<iostream>
 2 #include<set>
 3 #include<string>
 4 using namespace std;
 5 struct Info
 6 {
 7     string name;
 8     double score;
 9     bool operator < (const Info &a) const // 重载“<”操作符，自定义排序规则
10     {
11         //按score由大到小排序。如果要由小到大排序，使用“>”即可。
12         return a.score < score;
13     }
14 };
15 int main()
16 {
17     set<Info> s;
18     Info info;
19 
20     //插入三个元素
21     info.name = "Jack";
22     info.score = 80;
23     s.insert(info);
24     info.name = "Tom";
25     info.score = 99;
26     s.insert(info);
27     info.name = "Steaven";
28     info.score = 60;
29     s.insert(info);
30 
31     set<Info>::iterator it;
32     for(it = s.begin(); it != s.end(); it++)
33         cout << (*it).name << " : " << (*it).score << endl; 
34     return 0;
35 }