如何在C++中创建一个具有自定义比较器的std::set？

Question

我如何创建一个由成对元素组成的集合，其中这些成对元素通过自定义bool函数进行排序？我的写法是：

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

出现错误：参数2的类型/值不匹配，期望类型为“compare”，实际却是其他。

我已经将“compare”进行了定义。

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

当然了。

#include <vector>
#include <set>

- manolismi

Answer 1

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

- herohuyongtao

---

你也可以使用一个函数。你只需要正确获取类型即可。 - juanchopanza

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

- Vlad from Moscow