1. 程式人生 > >C++中關於set的自定義排序函式的書寫

C++中關於set的自定義排序函式的書寫

大概有兩個月沒用過C++啦,手都變得很生了,在這裡,在這裡我想扯一下關於set的比較函式的定義,我想,應該有不少人對這個東西感到頭疼吧!

如果說我想在set裡面新增一個自定義的型別,比如說下面的結構體:

struct Symbol
{
    char name[10];
};

我們如果要用set來盛放Symbol,我們應該怎麼辦麼?

set<Symbol> gSym;

如果像上面這麼來寫的話,或許你的編譯能夠通過,但是應該插入兩個元素之後就會報錯:

Symbol s1 = {"hi"};
Symbol s2 = {"girls"};
gSym.insert(s1);
gSym.insert(s2);

你這麼幹是不行的,C++編譯器操蛋的地方在於,它默默地為你幹了一些事情,我這裡稍微來剖析一下這段程式碼裡,編譯器到底為你幹了些什麼。

其實set的原型如下:

template <class Key,
          class Compare = less <key>,
          class Alloc = alloc>
class set {
...
};

模版裡面預設的比較函式是less,less函式的定義如下:

template <class T>
struct less : public binary_function<T, T, bool
> { bool operator () (const T& x, const T& y)const { return x < y; } };

less函式其實是一個仿函式,你在上面定義 set gSym 的時候,其實編譯器預設給你的比較函式是 less ,gSym要插入元素的時候,元素要和set裡面的元素進行比較,然而 x < y ,C++並不知道如何比較,x和y是Symbol型別,所以就報錯了。

如何解決呢?給Symbol型別定義一個比較函式即可:

struct Symbol
{
        char name[10];

        bool
operator< (const Symbol& r) const { if (strcmp(name, r.name) < 0) return true; else return false; } };

好了,修改程式碼如上面即可。現在你執行程式碼,應該沒問題了。

好吧,現在你想讓set中的元素從大到小排序,怎麼辦?

set<Symbol, greater<Symbol>> gSym;

然後我們給Symbol定義一個 > 即可。

當然,你也可以不用greater, less等仿函式,你可以自己定義仿函式,怎麼定義呢?仿照上面的less函式即可,去掉泛型。

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

struct Symbol
{
        char name[10];
};

struct SymCmp
{
        bool operator () (const Symbol& x, const Symbol& y) const
        {
                if (strcmp(x.name, y.name) < 0)
                        return true;
                else
                        return false;
        }
};


set<Symbol, SymCmp> gSet;

int main()
{
        Symbol s1 = { "hi" };
        Symbol s2 = { "girls" };

        gSet.insert(s1);
        gSet.insert(s2);
        gSet.insert(s1);
        gSet.insert(s2);
        gSet.insert(s1);

        system("pause");
        return 0;
}

好了,我想,我說清楚了,大家詳細瞭解的話,請檢視