【STL-set】

1. 所有元素会在插入时自动排序
2. 底层结构是二叉树
3. set和multiset区别.set不允许重复，multiset可以重复
4. bool compare()(int int) const 要加const
5. test08对于自定义数据类型，set必须指定排序规则才可以插入数据

#include<iostream>
#include<set>
#include<string>
using namespace std;


//打印输出
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;
	//set插入用insert，不用push，可能是因为set默认按顺序
	s1.insert(10);
	s1.insert(20);
	s1.insert(30);
	s1.insert(40);
	printset(s1);

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

	//赋值
	set<int>s3;
	s3 = s1;
	printset(s3);

}

//大小和交换
void test02()
{
	set<int> s1;
	//set插入用insert，不用push，可能是因为set默认按顺序
	s1.insert(10);
	s1.insert(20);
	s1.insert(30);
	s1.insert(40);
	printset(s1);

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

	set<int> s2;
	//set插入用insert，不用push，可能是因为set默认按顺序
	s2.insert(100);
	s2.insert(200);
	s2.insert(300);
	s2.insert(400);
	printset(s2);

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

	s1.swap(s2);  //交换

	cout << "交换后： " << endl;
	printset(s1);
	printset(s2);

}

//插入和删除
void test03()
{
	//insert(elem); //在容器中插入元素。
	//clear(); //清除所有元素
	//erase(pos); //删除pos迭代器所指的元素，返回下一个元素的迭代器。
	//erase(beg, end); //删除区间[beg,end)的所有元素 ，返回下一个元素的迭代器。
	//erase(elem); //删除容器中值为elem的元素。
	set<int> s1;
	//set插入用insert，不用push，可能是因为set默认按顺序
	s1.insert(10);
	s1.insert(20);
	s1.insert(30);
	s1.insert(40);
	printset(s1);

	//删除
	s1.erase(s1.begin());
	printset(s1);
	set<int>::iterator it = s1.begin();
	s1.erase(it); //索引
	printset(s1);

	//直接删除值
	s1.erase(30);
	printset(s1);

	s1.clear();
	printset(s1);

}

//查找和统计
void test04()
{
	//find(key); //查找key是否存在,若存在，返回该键的元素的迭代器；若不存在，返回set.end();
	//count(key); //统计key的元素个数
	set<int> s1;
	//set插入用insert，不用push，可能是因为set默认按顺序
	s1.insert(10);
	s1.insert(20);
	s1.insert(30);
	s1.insert(40);
	printset(s1);

	set<int>::iterator pos = s1.find(300);
	if (pos != s1.end()) {
		cout << "找到了元素：" << *pos << endl;
	}
	else
	{
		cout << "找不到" << endl;
	}

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

//用multiset插入同样的元素可以，set不可以
void test05()
{
	set<int> s;
	pair<set<int>::iterator, bool> ret = s.insert(10);
	if (ret.second) { //bool值为1执行   first 和 second 分别代表pair的
		cout << "第一次插入成功!" << endl;
	}
	else
	{
		cout << "第一次插入失败" << endl;
	}
	ret = s.insert(10);  //再插入一个同样的数
	if (ret.second)
	{
		cout << "第二次插入成功" << endl;
	}
	else
	{
		cout << "第二次插入失败" << endl;
	}

	multiset<int> ms;
	ms.insert(10);
	ms.insert(10);

	for (multiset<int>::iterator it = ms.begin(); it != ms.end(); it++)
	{
		cout << *it << " ";
	}
	cout << endl;
}

//pair对组创建
void test06()
{
	/*pair<type, type> p(value1, value2);
	pair<type, type> p = make_pair(value1, value2);*/
	set<int> s1;
	//set插入用insert，不用push，可能是因为set默认按顺序
	s1.insert(10);
	s1.insert(20);
	s1.insert(30);
	s1.insert(40);
	printset(s1);
	//pair<type, type> p(value1, value2);
	pair<string, int> p(string("Tom"), 20);
	cout << "姓名: " << p.first << " 年龄: " << p.second;

	//pair<type, type> p = make_pair(value1, value2)
	pair<string, int> p2 = make_pair("Tom", 20);
	cout << "姓名: " << p.first << " 年龄: " << p.second;
}

//set容器排序  利用仿函数，改变排序规则
class Mycompare
{
public:
	bool operator()(int v1, int v2) const
	{
		return v1 > v2; //逆序
	}
};
void test07()
{
	set<int> s1;
	//set插入用insert，不用push，可能是因为set默认按顺序
	s1.insert(10);
	s1.insert(20);
	s1.insert(30);
	s1.insert(40);
	printset(s1);

	for (set<int>::const_iterator it = s1.begin(); it != s1.end(); it++)
	{
		cout << *it << " ";
	}
	cout << endl;
	//指定排序规则
	set<int, Mycompare> s2;
	s2.insert(10);
	s2.insert(20);
	s2.insert(30);
	s2.insert(40);
	s2.insert(30);

	for (set<int,Mycompare>::iterator it = s2.begin(); it != s2.end(); it++)
	{
		cout << *it << " ";
	}
	cout << endl;
}

//存放自定义数据类型的排序
class Person
{
public:
	Person(string name, int age)
	{
		mname = name;
		mage = age;
	}
	string mname;
	int mage;
};

class	compareperson
{
public:
	bool operator()(const Person& p1, const Person& p2) const
	{
		return p1.mage > p2.mage;
	}
};
void test08()
{
	set<Person, compareperson> s;

	Person p1("刘备", 24);
	Person p2("关羽", 26);
	Person p3("张飞", 27);
	Person p4("赵云", 23);
	s.insert(p1);
	s.insert(p2);
	s.insert(p3);
	s.insert(p4);

	for (set<Person, compareperson>::iterator it = s.begin(); it != s.end(); it++)
	{
		cout << "姓名：" << it->mname << "年龄：" << it->mage << endl;
	}
}

int main()
{
	test01();
	test02();
	test03();
	test04();
	test05();
	test06();
	test07();
	test08();
	return 0;
}