set集合容器的妙用（用于去重，排序）

最新推荐文章于 2024-01-29 20:20:54 发布

原创最新推荐文章于 2024-01-29 20:20:54 发布 · 1.5k 阅读

7 ·

CC 4.0 BY-SA版权

C++ 专栏收录该内容

15 篇文章

订阅专栏

本文介绍如何在C++的set容器中实现自定义排序功能。通过两种方式：一是针对基本类型，自定义比较函数；二是对于结构体类型，在结构体内部重载小于运算符。这两种方法都能有效地改变set的默认排序行为。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

set中自定义比较函数
使用insert()将元素插入到集合中去的时候，集合会根据设定的比较函数将该元素放到该放的节点上去。在定义集合的时候，如果没有

指定比较函数，那么采用默认的比较函数(按键值又小到大的顺序插入元素)。

编写比较函数的两张方法：

1.如果元素不是结构体，那么可以编写比较函数。

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

//自定义比较函数myComp，重载操作符 ()  
struct myComp  
{  
    bool operator() (const int &a, const int &b)  
    {  
        return a > b;    //从大到小排序  
        //return a < b;  //从小到大排序  
    }  
};  

int main()  
{  
    set<int, myComp> s1;  
    for(int i = 1; i < 6; i++)  
        s1.insert(i*i);  
    s1.insert(8);  
    //ostream_iterator<int> output(cout, " ");  
    set<int, myComp>::iterator it = s1.begin();  
    for(; it != s1.end(); it++)  
    {  
        cout<<*it<<" ";  
    }  

    return 0;  
}

2.如果元素是结构体，那么可以直接把比较函数写在结构体内。

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

struct Info  
{  
    string name;  
    float score;  
    //重载操作符<,自定义排序规则  
    bool operator< (const Info &a)const  
    {  
        //return a.score < score;    //按score由大到小排列  
        return a.score > score;  //由小到大排列  
    }  
};  

int main()  
{  
    set<Info> s;  
    Info info;  
    info.name = "Messi";  
    info.score = 8.5;  
    s.insert(info);  
    info.name = "Ronae";  
    info.score = 9.0;  
    s.insert(info);  
    info.name = "My";  
    info.score = 7.0;  
    s.insert(info);  
    info.name = "Perno";  
    info.score = 8.0;  
    s.insert(info);  
    info.name = "Arzar";  
    info.score = 8.5;  
    s.insert(info);  
    set<Info>::iterator it;  
    for(it = s.begin(); it != s.end(); it++)  
    {  
        cout<<it->name<<" : ";  
        cout<<(*it).score<<endl;  
    }  
        return 0;  
}