常用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;