如何在 C++ 中使用自定义比较器创建 std::set？

Question

如何创建一组对，其中的元素（对）使用自定义布尔函数进行排序？我写

set <pair<int,int>,compare> myset;

并得到错误：参数 2 处的类型/值不匹配，需要一个类型，得到“比较”

我将“比较”定义为

bool compare(pair <int,int> g1, pair <int,int> g2)
{
    return (g1.second-g1.first > g2.second-g2.first);
}

当然

#include <vector>
#include <set>

Answer 1

方法一：使用函子

编写一个重载

operator()

的类，以便可以像函数一样调用它：

struct compare {
    bool operator() (const pair<int,int> &lhs, const pair<int,int> &rhs) const{
         return (lhs.second-lhs.first > rhs.second-rhs.first);
    }
};

然后，就可以使用类名作为类型参数了

set<pair<int,int>, compare> myset;

方法二：使用函数指针

假设

compare

是您要使用的功能：

set<pair<int,int>, bool(*)(const pair<int,int> &lhs, 
                           const pair<int,int> &rhs)
   > myset(&compare);

Answer 2

您应该使用函数对象。这是一个例子

#include <iostream>
#include <set>
#include <utility>

struct Compare
{
    bool operator ()( const std::pair<int, int> &p1, 
                      const std::pair<int, int> &p2 ) const
    {
        return ( p1.second - p1.first  > p2.second - p2.first );
    }
};

int main() 
{
    std::set<std::pair<int, int>, Compare> s;

    return 0;
}

Answer 3

从 C++ 20 开始，可以直接传递 lambda 类型。

set<pair<int, int>, decltype([](auto &g1, auto &g2){return g1.second-g1.first > g2.second-g2.first;})> myset;