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


105

我试图将一组整数中的项目的默认顺序更改为字典式而不是数字式,并且无法使用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

我究竟做错了什么?

Answers:


158

您正在使用应在其中使用函子的函数(重载()运算符的类,因此可以像函数一样调用它)。

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);

4
@Omry:我想知道您使用的是什么编译器:codepad.org/IprafuVf

1
@Omry您正在使用哪个编译器?

4
@Omry C ++标准说第二个模板参数必须是类型的名称-函数名称不是类型的名称。

6
我们可以使用decltype(lex_compare)表示函数类型吗?
刘易斯·陈

2
@LewisChan正确的用词是std::set<int64_t, decltype(&lex_compare)> s(&lex_compare)
Nishant Singh

109

1.现代C ++ 20解决方案

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

我们使用lambda函数作为比较器。与往常一样,比较器应返回布尔值,指示在第一个参数中作为第一个参数传递的元素是否被认为在第二个参数之前以其定义严格弱顺序

在线演示

2.现代C ++ 11解决方案

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

在C ++ 20之前,我们需要将lambda作为参数传递给set构造函数

在线演示

3.与第一个解决方案相似,但功能代替了lambda

使比较器像通常的布尔函数一样

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

然后以这种方式使用它:

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

在线演示

或者这样:

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

在线演示

4.使用带有()运算符的struct的旧解决方案

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

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

在线演示

5.替代解决方案:从布尔函数创建结构

采取布尔函数

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;

在线演示


3
在示例1中,是否需要将cmp传递给构造函数?集合会自行构造lambda类型作为模板类型吗?
PeteUK

2
@PeteUK必须在C ++ 20比较器之前传递到构造函数中。在C ++ 20中,可以使用不带参数的构造函数。谢谢你的提问;答案已更新
diralik

1
@diralik非常感谢您的答复并更新您已经很好的答案。
PeteUK

1
通用lambda似乎也适用于1和2
ZFY

2
那5.太疯狂了。人们每天都在寻找这种语言的新角落。
JanHošek

18

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!

哇,我觉得值得这么做!


17
我想是个观点问题。

6

您可以使用函数比较器而不必像这样包装它:

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

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

每当您需要该类型的集合时,这很烦人,并且如果您不使用相同的比较器来创建所有集合,则会引起问题。


3

std::less<> 当使用自定义类 operator<

如果您要处理一组已operator<定义的自定义类,则可以使用std::less<>

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

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

这使您可以:

main.cpp

#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);
}

编译并运行:

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

有关更多信息,请std::less<>参见:什么是透明比较器?

已在Ubuntu 16.10、6.2.0上测试g++

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.