C++中關於set的自定義排序函式的書寫
阿新 • • 發佈:2019-02-19
大概有兩個月沒用過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;
}
好了,我想,我說清楚了,大家詳細瞭解的話,請檢視