set/multiset容器

set/multiset容器

1.set/multiset特性

set/multiset特性是所有元素都会根据元素的值进行自动排序。set是以RB-tree（红黑树，平衡二叉树的一种）为底层机制，其查找效率非常好。
set容器中不允许重复元素，multiset允许重复元素。

2.set/multiset常用API

2.1.构造函数

//set默认构造函数
set<T> st;
//multiset默认构造函数
multiset<T> mst;
//拷贝构造函数
set(const set &st);

2.2.赋值操作

//重载等号运算符
set& opreator=(const set &st);
//交换两个集合容器
swap(st);

2.3.大小操作

//返回容器中元素数目
size();
//判断容器是否为空
empty();

2.4.插入和删除操作

//在容器中插入元素
insert(elem);
//清除所有元素
clear();
//删除pos迭代器所指的元素，返回下一个元素的迭代器
erase(pos);
//删除区间[beg,end)的所有元素，返回下一个元素的迭代器
erase(beg, end);
//删除容器中值为elem的元素,set中不允许元素重复，所以没有重复的元素，只删除单个元素；
//multiset允许有重复的元素，删除的是所有值为elem的元素
erase(elem);

2.4.1.插入删除举例

#include <iostream>
#include <set> //set/multiset需要包含的头文件
using namespace std;

void print_multiset(multiset<int>& mset,int flag)
{
	cout<<"print["<<flag<<"]：";
	for(multiset<int>::iterator it = mset.begin();it != mset.end();it++)
	{
		cout<<(*it)<<" ";
	}
	cout<<endl;
}

int main()
{
	multiset<int> mset;
	mset.insert(6);
	mset.insert(5);
	mset.insert(7);
	mset.insert(9);
	mset.insert(6);
	mset.insert(1);
	mset.insert(3);
	mset.insert(8);
	mset.insert(6);
	print_multiset(mset,1);
	mset.erase(6);
	print_multiset(mset,2);
	
	return 0;
}

执行结果：

root@host:/home/LinuxShare/007.STL/set# g++ 01_del.cpp
root@host:/home/LinuxShare/007.STL/set# ./a.out
print[1]：1 3 5 6 6 6 7 8 9
print[2]：1 3 5 7 8 9

2.4.2.排序规则(仿函数)举例

如果set中元素是int型数据，set默认从小到大排序。但是如果set中插入的是对象呢？
对象中有多个成员变量，set怎么知道根据哪个成员变量排序？
可以通过仿函数告诉其排序规则。

#include <iostream>
#include <set> //set/multiset需要包含的头文件
using namespace std;

class Persion{
public:
	Persion(int age,int id):id(id),age(age){}
public:
	int age;
	int id;
};
//仿函数
class mycompare{
public:
	bool operator()(Persion p1, Persion p2)
	{
		return p1.age > p2.age;
	}
};

int main()
{
	//通过仿函数告诉set中数据的排序规则
	set<Persion,mycompare> sp;
	Persion p1(10,20),p2(30,40),p3(50,60);
	sp.insert(p1);
	sp.insert(p2);
	sp.insert(p3);

	for (set<Persion,mycompare>::iterator it = sp.begin(); it != sp.end();it++)
	{
		cout<<(*it).age<<" "<<(*it).id<<endl;
	}
	return 0;
}

执行结果：

root@host:/home/LinuxShare/007.STL/set# g++ 04_paixu.cpp
root@host:/home/LinuxShare/007.STL/set# ./a.out
50 60
30 40
10 20

2.5.set查找操作

//查找键key是否存在，若存在，返回该键的元素的迭代器；若不存在，返回st.end()
//multiset容器中返回的是最后一个键值key的元素的迭代器
find();
//返回第一个>=keyElem元素的迭代器
lower_bound(keyElem);
//返回第一个>keyElem元素的迭代器
upper_bound(keyElem);
//返回容器中与keyElem相等的上下限的两个迭代器
//返回pair，分别为lower_bound和upper_bound两个迭代器
equal_range(keyElem);

2.5.1.find函数查找举例1

#include <iostream>
#include <set> //set/multiset需要包含的头文件
using namespace std;

class Persion{
public:
	Persion(int age,int id):id(id),age(age){}
public:
	int age;
	int id;
};
//仿函数
class mycompare{
public:
	bool operator()(Persion p1, Persion p2)
	{
		return p1.age > p2.age;
	}
};

int main()
{
	set<Persion,mycompare> sp;
	Persion p1(10,20),p2(30,40),p3(50,60);
	sp.insert(p1);
	sp.insert(p2);
	sp.insert(p3);
	
	for (set<Persion,mycompare>::iterator it = sp.begin(); it != sp.end();it++)
	{
		cout<<(*it).age<<" "<<(*it).id<<endl;
	}
	//查找,
	//虽然要找的是age=10,id=30,sp中没有
	//但是由于是按照age排序,实际中要找到age=10的就算找到
	Persion p4(10,30);
	set<Persion,mycompare>::iterator it = sp.find(p4);
	if (it == sp.end())
	{
		cout<<"没找到"<<endl;
	}
	else
	{
		cout<<"找到："<<(*it).age<<","<<(*it).id<<endl;
	}
	
	return 0;
}

执行结果：

root@host:/home/LinuxShare/007.STL/set# g++ 04_paixu.cpp
root@host:/home/LinuxShare/007.STL/set# ./a.out
50 60
30 40
10 20
找到：10,20

2.5.2.find函数查找举例2

#include <iostream>
#include <set> //set/multiset需要包含的头文件
using namespace std;

void print_multiset(multiset<int>& mset,int flag)
{
	cout<<"print["<<flag<<"]：";
	for(multiset<int>::iterator it = mset.begin();it != mset.end();it++)
	{
		cout<<(*it)<<" ";
	}
	cout<<endl;
}

int main()
{
	multiset<int> mset;
	mset.insert(6);
	mset.insert(5);
	mset.insert(7);
	mset.insert(9);
	mset.insert(9);
	mset.insert(1);

	print_multiset(mset,1);
	multiset<int>::iterator ret = mset.find(9);
	if (ret == mset.end())
	{
		cout<<"没找到"<<endl;
	}
	else
	{
		cout<<"ret:"<<*ret<<endl;
	}
	
	return 0;
}

执行结果：

root@host:/home/LinuxShare/007.STL/set# g++ 02_find.cpp
root@host:/home/LinuxShare/007.STL/set# ./a.out
print[1]：1 5 6 7 9 9
ret:9

2.5.3.equal_range函数举例

#include <iostream>
#include <set> //set/multiset需要包含的头文件
using namespace std;

void print_multiset(multiset<int>& mset,int flag)
{
	cout<<"print["<<flag<<"]：";
	for(multiset<int>::iterator it = mset.begin();it != mset.end();it++)
	{
		cout<<(*it)<<" ";
	}
	cout<<endl;
}

int main()
{
	multiset<int> mset;
	mset.insert(6);
	mset.insert(5);
	mset.insert(7);
	mset.insert(9);
	mset.insert(6);
	mset.insert(1);
	mset.insert(3);
	mset.insert(8);
	mset.insert(6);
	print_multiset(mset,1);
	pair<multiset<int>::iterator,multiset<int>::iterator> ret = mset.equal_range(6);
	if (ret.first == mset.end())
	{
		cout<<"没找到"<<endl;
	}
	else
	{
		cout<<"ret:"<<*(ret.first)<<endl;
	}
	if (ret.second == mset.end())
	{
		cout<<"没找到"<<endl;
	}
	else
	{
		cout<<"ret:"<<*(ret.second)<<endl;
	}
	return 0;
}

执行结果：

root@host:/home/LinuxShare/007.STL/set# g++ 03_equal_range.cpp
root@host:/home/LinuxShare/007.STL/set# ./a.out
print[1]：1 3 5 6 6 6 7 8 9
ret:6
ret:7