常用STL使用指北

常用STL使用指北

set和multiset

set和multiset都是基于红黑树（显然是一个二叉搜索树）的STL。

定义

我们可以使用(multi)set<元素类型>名称来定义一个(multi)set。

自定义排序

默认排序方式都是从小到大。因为结构体之间没有定义<，所以我们需要自定义一个比较函数。

如果元素不是结构体：

//自定义比较函数myComp,重载“（）”操作符
struct myComp
{
	bool operator()(const int &a,const int &b)
	{
		return a > b;//从大到小排序
	}
}
set<int,myComp>s;
set<int,myComp>::iterator it;
 

如果元素是结构体：

struct Info
{
	string name;
	float score;
	//重载“<”操作符，自定义排序规则
	bool operator < (const Info &a) const
	{
		//按score从大到小排列
		return a.score<score;
	}
}
 
set<Info> s;
set<Info>::iterator it;

或者

//自定义比较函数myComp,重载“（）”操作符
struct myComp
{
	bool operator()(const your_type &a,const your_type &b)
	{
		return a.data > b.data;//按照data从大到小排序
	}
}
set<your_type,myComp>s;
set<your_type,myComp>::iterator it;