如何在C ++中使用自定义排序成员函数的sort()?

时间:2015-03-26 18:42:25

标签: c++ sorting functor

我有一个关于将比较函数传递给sort()的问题。

我想要做的是定义一个sort()函数,该函数在计算时考虑了我想要进行排序的类的成员变量。

基本上,我的代码看起来像这样(简化为只显示相关部分):

MappingTechnique.h

struct MappingTechnique {
    vector<int> usedIndexCount; 
};

struct SimpleGreedyMappingTechnique : MappingTechnique {
    bool sortByWeights(int index1, int index2);
};

MappingTechnique.m

bool SimpleGreedyMappingTechnique::sortByWeights(int index1, int index2) {
    return usedIndexCount[index1] > usedIndexCount[index2];
}

void SimpleGreedyMappingTechnique::processFrame(Frame frame) {
    vector<int> payloadIndices = <generate the vector>

    // sort the payload indices according to their current usedIndexCount
    sort(payloadIndices.begin(), payloadIndices.end(), sortByWeights);
}

此代码无法编译,它会出现以下错误:

 error: reference to non-static member function must be called

并指向sortByWeights

甚至可以使用类的成员函数进行排序吗?如果是,我该如何实现呢?

2 个答案:

答案 0 :(得分:6)

确实如此,但总的来说,我鼓励使用合适的仿函数或lambda:

使用lambda:

std::sort(payloadIndices.begin(), payloadIndices.end(), [this](int a, int b){
    return this->sortByWeights(a, b);
});

或者使用std::mem_fn

auto sorter = std::bind(std::mem_fn(SimpleGreedyMappingTechnique::sortByWeights), this);
std::sort(payloadIndices.begin(), payloadIndices.end(), sorter);

或者使用仿函数:

namespace{
struct indicies_less_than
{
    const SimpleGreedyMappingTechnique & mapping_tech;
    indicies_less_than(const SimpleGreedyMappingTechnique & mapping_tech)
       :mapping_tech(mapping_tech){}

    bool operator()(int a, int b)
    {
       return mapping_tech.sortByWeights(a, b);
    }
};
}

std::sort(payloadIndices.begin(), payloadIndices.end(), indicies_less_than(*this));

注意:

如果要排序的类型比int更复杂,您肯定希望通过const&传递它们以防止复制

答案 1 :(得分:2)

正如Mgetz所提到的,可以使用仿函数。仿函数的例子:

#include <algorithm>
#include <cstdlib>
#include <iostream>
#include <iomanip>
#include <vector>

#define asize 16

class example{
public:
    unsigned int a[asize];      // array
    std::vector<size_t> v;      // indices to array

example(void)
{
    v.resize(asize);
    for(size_t i = 0; i < asize; i++){
        a[i] = rand()%10;
        v[i] = i;
    }
}

void displayarray(void)
{
    for(size_t i = 0; i < asize; i++)
        std::cout << std::setw(3) << a[v[i]];
    std::cout << std::endl;
}

class lessthan                  // lessthan functor for std::sort
{
public:
const example &x;
    lessthan(const example &e ) : x(e) { }
    bool operator()(const size_t & i0, const size_t & i1)
    {
        return x.a[i0] < x.a[i1];
    }
};

void sortarray(void)
{
    std::sort(v.begin(), v.end(), lessthan(*this));
}
};

int main()
{
example x;

    x.displayarray();
    x.sortarray();
    x.displayarray();
    return 0;
}