C++之set自定義排序問(wèn)題

更新時(shí)間：2023年11月08日 09:20:14 作者：滄海漂游_

這篇文章主要介紹了C++之set自定義排序問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

set簡(jiǎn)介

set一般插入元素時(shí)，默認(rèn)使用關(guān)鍵字類(lèi)型的< 運(yùn)算符來(lái)比較兩個(gè)關(guān)鍵字，

故一般插入后為升序，但是針對(duì)自定義數(shù)據(jù)結(jié)構(gòu)，

如結(jié)構(gòu)體，沒(méi)有< 運(yùn)算符，故無(wú)法進(jìn)行比較。

針對(duì)自定義數(shù)據(jù)結(jié)構(gòu)或者說(shuō)自定義set排序規(guī)則有如下幾種方法：

方法一重載<

在自定義結(jié)構(gòu)體中重載< 則可以實(shí)現(xiàn)默認(rèn)排序，

示例代碼如下：

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

struct Students
{
    string id;
    int age,height;
    Students(string s,int a,int h):id(s),age(a),height(h){}
    Students() {}
    bool operator <(const Students &s) const {
        if(id!=s.id) return id<s.id;
        else return age<s.age;
    }
};
int main(){
    set<Students> se;
    se.insert(Students("zhou",12,134));
    se.insert(Students("wu",13,42));
    se.insert(Students("zheng",34,43));
    se.emplace("wang",13,43);
    se.emplace("zhou",23,43);
    for(auto it=se.begin();it!=se.end();it++){
        cout<<it->id<<" "<<it->age<<" "<<it->height<<endl;
    }
    return 0;
}

運(yùn)行結(jié)果如下：

這里寫(xiě)圖片描述

方法二重載()

示例代碼如下：

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

struct Students
{
    string id;
    int age,height;
    Students(string s,int a,int h):id(s),age(a),height(h){}
    Students() {}
};

class comp{
public:
    bool operator()(const Students &s1,const Students &s2){
        if(s1.id!=s2.id) return s1.id<s2.id;
        return s1.age<s2.age;
    }
};

int main(){
    set<Students,comp> se;
    se.insert(Students("zhou",12,134));
    se.insert(Students("wu",13,42));
    se.insert(Students("zheng",34,43));
    se.emplace("wang",13,43);
    se.emplace("zhou",23,43);
    for(auto it=se.begin();it!=se.end();it++){
        cout<<it->id<<" "<<it->age<<" "<<it->height<<endl;
    }
    return 0;
}

方法三參考《C++ primer（第五版）》

示例代碼如下：

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

struct Students
{
    string id;
    int age,height;
    Students(string s,int a,int h):id(s),age(a),height(h){}
    Students() {}
};

bool cmp(const Students &s1,const Students &s2){
    if(s1.id!=s2.id) return s1.id<s2.id;
    return s1.age<s2.age;
}

int main(){
    set<Students,decltype(cmp)*> se(cmp);
    se.insert(Students("zhou",12,134));
    se.insert(Students("wu",13,42));
    se.insert(Students("zheng",34,43));
    se.emplace("wang",13,43);
    se.emplace("zhou",23,43);
    for(auto it=se.begin();it!=se.end();it++){
        cout<<it->id<<" "<<it->age<<" "<<it->height<<endl;
    }
    return 0;
}

上述代碼中，用decltype 來(lái)指出自定義操作的類(lèi)型。

當(dāng)使用decltype 來(lái)獲得一個(gè)函數(shù)指針類(lèi)型時(shí)，必須加上一個(gè)* 來(lái)指出我們要使用一個(gè)給定函數(shù)類(lèi)型的指針。

用cmp 來(lái)初始化se對(duì)象，這表示當(dāng)我們向se中插入元素時(shí)，通過(guò)調(diào)用cmp來(lái)為這些元素排序。

可以使用cmp代替&cmp作為構(gòu)造函數(shù)的參數(shù)，因?yàn)楫?dāng)我們使用一個(gè)函數(shù)的名字時(shí)，在需要的情況下會(huì)自動(dòng)轉(zhuǎn)化為一個(gè)指針，使用&cmp 效果也是一樣的。