如果说我想在set里面添加一个自定义的类型,比如说下面的结构体:
struct Symbol
{
char name[10];
};
我们如果要用set来盛放Symbol,我们应该怎么办么?
set<Symbol> gSym;
- 1
如果像上面这么来写的话,或许你的编译能够通过,但是应该插入两个元素之后就会报错:
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;
- 1
然后我们给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;
}