使用自定义比较器设置的 C++ STL

Question

对于 C++ STL set，我需要自定义比较，但我的代码存在严重问题。使用 Visual Studio 2019 编译失败。第一个错误是（在 xutility 中）“错误 C2672: 'operator __surrogate_func': 找不到匹配的重载函数”。

代码如下：

#include <set>

struct P
{
    char a;
    char b;
    char c;
};

class PSortingCriterion
{
public:
    bool operator() (const P &p1, const P &p2) const
    {
        if(p1.a != p2.a)
            return p1.a < p2.a;
        else if(p1.b != p2.b)
            return p1.b < p2.b;
        else if(p1.c != p2.c)
            return p1.c < p2.c;
        else
            return false; // p1 == p2
    }
};

std::set<P, PSortingCriterion> s1, s2, diff;

void F(void)
{
    if(s1 != s2) // errors are all due to this statement; no errors if it's removed
    {
    }
}

更新：按照 Igor Tandetnik 的建议，我修改了代码，但仍然无法编译（与上述相同的错误）。

新代码：

#include <set>
#include <algorithm>
#include <iterator>

struct P
{
    char a;
    char b;
    char c;
};

class PSortingCriterion
{
public:
    bool operator() (const P &p1, const P &p2) const
    {
        if(p1.a != p2.a)
            return p1.a < p2.a;
        else if(p1.b != p2.b)
            return p1.b < p2.b;
        else if(p1.c != p2.c)
            return p1.c < p2.c;
        else
            return false; // p1 == p2
    }
};

std::set<P, PSortingCriterion> s1, s2, diff;

void F(void)
{
    set_difference(s1.begin(), s1.end(), s2.begin(), s2.end(), inserter(diff, diff.begin()));
}


   

Answer 1

要比较两个集合，您需要比较元素。见CppReference。但是您没有提供任何方法来比较两个 P 类型的对象是否相等。

如果你添加：

bool operator== (const P& p1, const P& p2) {
   return p1.a == p2.a && p1.b == p2.b && p1.c == p2.c;
}

然后您的代码将编译。

Answer 2

这是一个重载 == 和 != 运算符的简单示例（我认为这就是你想要做的事情？）

#include<set>
#include<iostream>

struct P {
    char a;
    char b;
    char c;
    bool operator==(const P &p2) const {
      return this->a == p2.a && this->b == p2.b && this->c == p2.c;
    }
    bool operator!=(const P &p2) const {
      return this->a != p2.a || this->b != p2.b || this->c != p2.c;
    }
};

void EqualExample(void) {
  P s1;
  s1.a = 'a';
  s1.b = 'b';
  s1.c = 'c';
  P s2;
  s2.a = 'a';
  s2.b = 'b';
  s2.c = 'c';
  if(s1 != s2){
    std::cout << "Not Equal" << std::endl;
  } else {
    std::cout << "Equal" << std::endl;
  }
}

void NotEqualExample(void) {
  P s1;
  s1.a = 'a';
  s1.b = 'b';
  s1.c = 'c';
  P s2;
  s2.a = 'a';
  s2.b = 'b';
  s2.c = 'd';
  if(s1 != s2){
    std::cout << "Not Equal" << std::endl;
  } else {
    std::cout << "Equal" << std::endl;
  }
}

int main () {
  EqualExample();
  NotEqualExample();
}