博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

C++中定义比较函数的三种方法

Posted on 2013-12-25 17:59  lengbingshy  阅读(15515)  评论(1编辑  收藏  举报

原文地址:http://fusharblog.com/3-ways-to-define-comparison-functions-in-cpp/

 

C++编程优与Pascal的原因之一是C++中存在STL(标准模板库)。STL存在很多有用的方法。

C++模板库中的许多方法都需要相关参数有序,例如Sort()。显然,如果你想对一个集合进行排序,你必须要知道集合中的对象,那个在前那个在后。因此,学会如何定义比较方法是非常重要的。

C++模板库的许多容器需要相关类型有序,例如set<T> 和priority_queue<T>。

这篇文章旨在告诉大家如何为一个类定义一个排序方法,以便在STL容器或者方法中使用。 作为一个C++程序员,你应该知道这些方法。

 

如何定义排序?

简而言之,为一个类定义排序,我们就可以知道类的任意两个对象在排序的过程中谁在前谁在后。我们可以用一个方法来实现,这个方法返回一个bool值表示谁排在前面。显然,我们希望实现一个类似,f(x,y),这种形式的方法。它接收同一类型的对象作为两个参数,返回值则表明谁会出现在谁前面。

 

严格弱序化

几乎所有的方法或容器都需要排序来满足数学意义上的标准严格弱序化,否则这些方法或容器的行为将不可预知。

假设f(x,y)是一个比较函数。 如果该函数满足如下条件则它是严格弱序化的。

 

1.f(x,x) = false; 

2. if f(x,y) then !f(y,x)

3.if f(x,y) and f(y,z) then f(x,z)

4. if !f(x,y)&&!f(y,x) then x==y; if x==y and y==z then x==z;

看上去有点晕乎,不过不用担心,只要你的比较方法能够满足对相等元素永远返回false,那你的方法就满足要求了。

 

三种实现方式:

1. 定义 < 操作符。

 使用这种方法可以使我们自定义的类能够获得与生俱来的排序能力。例如,如果有如下类:

struct Edge
{
int from,to ,weight;
};

 因为要实现Kruskai算法,你希望图中的所有边依据权重按降序排列。 像这样来定义 operator<:

struct Edge
{
        int from,to ,weight;
        bool operator <(Edge other) const
        {
                return weight>other.weight;
        }       
};    

 你定义的方法必须按照如下方法声明:

bool operator< (T other) const

注意: const关键字是必须的。

如果你不喜欢这种方式,比如,明明是要比较两个对象,方法却只有一个参数。你可以选择如下方式:

 

struct Edge
{
  int from,to weight;
  friend bool operator<(Edge a,Edge b)
  {
    return a.weight>b.weight;
  }
};

 

STL的pair<T1,T2>就具有与生俱来的排序能力。两个pair对象的比较这样的:先比较第一个参数,如果第一个参数相同再比较第二个参数。

所有内置类型都具有与生俱来的排序能力,这是由编译器赋予的。

 

2. 自定义排序方法。

使用这种方式常常用在如下情形:

a.比较内置类型

b.不能修改需要比较的类型

c.除了类型自定义的比较方式以外的比较方法

简单来说,一个比较方法接收两个同类型的对象作为参数并且返回一个bool值,原型如下:

bool name(T a,T b);

 

3. 重载()操作符

我们可以将比较函数作为STL容器构造函数的第一个参数,并且把函数类型作为模板参数。例如:

set<int,bool (*)(int,int)> s(cmp);

 这样做或多或少会让人费解。那我们就来看看如何使用仿函数来消除你的疑惑吧。

我们需要定义一个新的类并重载()操作符。

vector<int> occurrences; 
struct cmp 
{ 
    bool operator()(int a, int b) 
    { 
        return occurrences[a] < occurrences[b];
    } 
}; 

现在我们就可以把这个类作为模板参数传递给STL容器了。

set<int, cmp> s; 

priority_queue<int, vector<int>, cmp> pq; 

 

 

STL也有一些内置的仿函数,例如less<T>,greater<T>等。

仿函数可以通过初始化然后像普通函数一样使用。最简单的就是在仿函数后面加上()。

sort(data.begin(), data.end(), greater<int>());