Yihulee's world!

========>易燃又美味!

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<Symbol> gSym 的时候,其实编译器默认给你的比较函数是 less<Symbol> ,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;
}

好了,我想,我说清楚了,大家详细了解的话,请查看<STL源码剖析>。

Comments