使用自定义比较函数进行c ​​++排序

时间:2016-06-20 22:02:55

标签: c++ sorting

我有以下类型的矢量:

std::vector< std::pair< std::pair< int, int >, std::vector<float> > >  neighbors;

我想创建以下向量排序如下

std::sort( neighbors.begin(), neighbors.end(), myFunc(index) );

其中,

bool myFunc( const std::pair< std::pair< int, int >, float > &a, const std::pair< std::pair< int, int >, float > &b, const int index ){
     return ( a.second[index] > b.second[index] );
}

我知道语法错误,但我希望提供一个仅用于比较矢量元素的索引。

我不知道如何将此参数传递给myFunc。

2 个答案:

答案 0 :(得分:0)

Lambdas:

std::sort(
    neighbors.begin(),
    neighbors.end(),
    [index](const std::pair< std::pair< int, int >, std::vector<float> > &a,  
            const std::pair< std::pair< int, int >, std::vector<float> > &b)
            { 
                 return ( a.second[index] > b.second[index] );
            }
);  

有关详细介绍,请参阅What is a lambda expression in C++11?

答案 1 :(得分:0)

Pre-C ++ 11解决方案:使用对象实例作为自定义比较器

struct Comparator {
    Comparator(int index) : index_(index) {}
    bool operator () (const std::pair< std::pair< int, int >, std::vector<float> > &a,
                      const std::pair< std::pair< int, int >, std::vector<float> > &b) 
    {
        return ( a.second[index_] > b.second[index_] );
    }

    int index_;
};

sort(neighbors.begin(), neighbors.end(), Comparator(42));

C ++ 11 +解决方案:使用lambda

std::sort(neighbors.begin(), neighbors.end(), [index]
                         (const std::pair< std::pair< int, int >, std::vector<float> > &a, 
                          const std::pair< std::pair< int, int >, std::vector<float> > &b) 
  { 
    return ( a.second[index] > b.second[index] );
  }
);

我的建议:如果允许使用C ++ 11功能,请选择lambda。它可能更优雅,更易读。