具有自定义结构的 <set> 包含重复项

Question

我一直在学习 C++。我被这个问题困住了。

我有 set，其中包含一个自定义结构，该结构包含两个 long int 的 a 和 b。我有一个自定义比较器结构，它比较数字并在 a 或 b 不同时返回 true。

typedef long int li;

struct number {
    number(li a1,li b1): a(a1), b(b1) {}
    li a, b;
};

struct compare {
    bool operator() (const number &lhs, const number& rhs) const{
        return lhs.a != rhs.a || lhs.b != rhs.b;
    }
};

int main() {
    set<number, compare> nums;
    nums.insert(number(1, 2));
    nums.insert(number(1, 1));
    nums.insert(number(2, 1));
    nums.insert(number(1, 2));
    for (auto &i : nums) {
        cout << i.a << " " << i.b << endl;
    }
    return 0;
}

这里的输出是

1 2

2 1

1 1

1 2

它有两个1 2条目。任何澄清将不胜感激。

Answer 1

您的比较函数应该返回某个元素是否小于另一个元素，而不是它们是否相等。 （更正式地说，它必须在你的集合元素上定义一个"Strict weak ordering"。）

使用类似的东西

struct compare {
    bool operator() (const number &lhs, const number& rhs) const{
        return std::tie(lhs.a, lhs.b) < std::tie(rhs.a, rhs.b);
    }
};

如果您不关心排序，您可能需要为您的类型定义一个合适的哈希函数并使用std::unordered_set。

为避免将来出现此类问题，请务必阅读the docs。他们清楚地解释了您的comparison function 应该做什么。

供参考：上面使用的std::tie 构造了对其参数的引用元组，然后可以按字典顺序与< 进行比较。这是一种简单、通用且快速的方法，可以为不具可比性的东西的集合建立一些排序。

Answer 2

您的比较函数需要满足严格/弱排序要求。

（我实际上更喜欢使用std::tie 的答案，但这对新手来说可能更能说明问题）

bool compare(const number& lhs, const number& rhs)
{
   if(lhs.a < rhs.a)
      return true;
   else if(lhs.a > rhs.a)
      return false;
   else
      return lhs.b < rhs.b;
}