c++ - 使用相同的函数对向量进行排序和自定义比较器

Question

这听起来像是一个愚蠢的问题，但我想了很长时间，有没有更好的方法：

struct X
{
    int a;
    int b;
};
bool sortComp(const X first, const X second)
{
    if (first.a!=second.a)
        return (first.a<second.a);
    else
        return (first.b<second.b);

}
class setComp
{
public:
    bool operator() (const X first, const X second) const
    {
        if (first.a!=second.a)
                return (first.a<second.a);
            else
                return (first.b<second.b);
    }
};
int main()
{
    vector<X> v;
    set<X, setComp> s;
    sort(begin(v), end(v),sortComp);
}

如您所见，我两次实现了相同的功能，一次用于排序，一次用于集合中的隐式排序。有没有办法避免代码重复？

Answer 1

当然，只需选择两者之一并更改另一个的调用。

// choosing the function object
sort(begin(v), end(v), setComp()); // create setComp, sort will call operator()

// choosing the function
set<X, bool(*)(const X, const X)> s(sortComp); // pass function pointer

我个人会推荐仿函数版本。