使用自定义std :: set比较器

Question

我试图将一组整数中的项的默认顺序更改为lexicographic而不是numeric，并且我无法使用g ++编译以下内容：

file.cpp：

bool lex_compare(const int64_t &a, const int64_t &b) 
{
    stringstream s1,s2;
    s1 << a;
    s2 << b;
    return s1.str() < s2.str();
}

void foo()
{
    set<int64_t, lex_compare> s;
    s.insert(1);
    ...
}

我收到以下错误：

error: type/value mismatch at argument 2 in template parameter list for ‘template<class _Key, class _Compare, class _Alloc> class std::set’
error:   expected a type, got ‘lex_compare’

我做错了什么？

Answer 1

你正在使用一个函数，你应该使用一个仿函数（一个重载（）运算符的类，因此可以像函数一样调用它。）

struct lex_compare {
    bool operator() (const int64_t& lhs, const int64_t& rhs) const {
        stringstream s1, s2;
        s1 << lhs;
        s2 << rhs;
        return s1.str() < s2.str();
    }
};

然后使用类名作为类型参数

set<int64_t, lex_compare> s;

如果你想避免仿函数样板代码，你也可以使用函数指针（假设lex_compare是一个函数）。

set<int64_t, bool(*)(const int64_t& lhs, const int64_t& rhs)> s(&lex_compare);

Answer 2

1。现代C ++ 11解决方案

auto cmp = [](int a, int b) { return ... };
std::set<int, decltype(cmp)> s(cmp);

我们使用lambda function作为比较器。像往常一样，比较器应该返回布尔值，指示作为第一个参数传递的元素是否被认为是在它定义的特定strict weak ordering中的第二个参数之前。

Online demo

2。与第一个解决方案类似，但使用函数而不是lambda

将比较器设为通常的布尔函数

bool cmp(int a, int b) {
    return ...;
}

然后使用它

std::set<int, decltype(&cmp)> s(&cmp);

Online demo

3。使用带()运算符

的struct的旧解决方案

struct cmp {
    bool operator() (int a, int b) const {
        return ...
    }
};

// ...
// later
std::set<int, cmp> s;

Online demo

4。替代解决方案：从布尔函数

创建结构

采用布尔函数

bool cmp(int a, int b) {
    return ...;
}

使用std::integral_constant

从中创建结构

#include <type_traits>
using Cmp = std::integral_constant<decltype(&cmp), &cmp>;

最后，使用struct作为比较器

std::set<X, Cmp> set;

Online demo

Answer 3

Yacoby的回答激励我编写一个用于封装仿函数样板的适配器。

template< class T, bool (*comp)( T const &, T const & ) >
class set_funcomp {
    struct ftor {
        bool operator()( T const &l, T const &r )
            { return comp( l, r ); }
    };
public:
    typedef std::set< T, ftor > t;
};

// usage

bool my_comparison( foo const &l, foo const &r );
set_funcomp< foo, my_comparison >::t boo; // just the way you want it!

哇，我认为那值得一试！

Answer 4

您可以使用功能比较器，而不像这样包装它：

bool comparator(const MyType &lhs, const MyType &rhs)
{
    return [...];
}

std::set<MyType, bool(*)(const MyType&, const MyType&)> mySet(&comparator);

每次需要一组该类型时输入都会很烦人，如果不使用相同的比较器创建所有集合，则会导致问题。

Answer 5

std::less<>与operator<一起使用自定义类

如果要处理一组定义了operator<的自定义类，则可以只使用std::less<>，例如：

如http://en.cppreference.com/w/cpp/container/set/find所述，C ++ 14添加了两个新的find API：

template< class K > iterator find( const K& x );
template< class K > const_iterator find( const K& x ) const;

允许您执行以下操作：

#include <cassert>
#include <set>

class Point {
    public:
        // Note that there is _no_ conversion constructor,
        // everything is done at the template level without
        // intermediate object creation.
        //Point(int x) : x(x) {}
        Point(int x, int y) : x(x), y(y) {}
        int x;
        int y;
};
bool operator<(const Point& c, int x) { return c.x < x; }
bool operator<(int x, const Point& c) { return x < c.x; }
bool operator<(const Point& c, const Point& d) {
    return c.x < d;
}

int main() {
    std::set<Point, std::less<>> s;
    s.insert(Point(1, -1));
    s.insert(Point(2, -2));
    s.insert(Point(0,  0));
    s.insert(Point(3, -3));
    assert(s.find(0)->y ==  0);
    assert(s.find(1)->y == -1);
    assert(s.find(2)->y == -2);
    assert(s.find(3)->y == -3);
    // Ignore 1234, find 1.
    assert(s.find(Point(1, 1234))->y == -1);
}

已在Ubuntu 16.10 g++ 6.2.0上进行了测试，并具有：

g++ -std=c++14 -Wall -Wextra -pedantic -o main.out main.cpp
./main.out

有关std::less<>的更多信息，请访问：What are transparent comparators?