c++ 重载运算符 () 更改 std::set.find()

Question

我正在学习 std::set 。这是我的代码：

#include <set>
#include <iostream>
using namespace std;
struct cmp{
    bool operator () (const int & a,const int & b) const {
        if(abs(a-b)<=3)
            return false;
        return a < b;
    }
};
set<int,cmp> q{1, 2, 10};
int main(){
    if(q.find(4)!=q.end())
        cout << 1;
    else
        cout << 2;
}

输出：1

我使用 struct cmp 来自定义元素的排序规则，如果 abs(a-b)<=3 新元素将被删除。

但是令我惊讶的是 q.find() 已经被改变了。

我想知道为什么输出是1，q中没有4。

q.find(4)是获取元素等于的迭代器 4 在 q 中，不是吗？

Answer 1

输出为 1 因为您的自定义比较器函数不仅用于对集合中的元素进行排序，而且还用于确定元素之间的相等性。换句话说，find 函数使用比较器来确定元素是否等于您要搜索的值（在本例中为 4）。

当 find 函数遇到集合中的元素 1 时，它会使用您的自定义比较器函数将其与 4 进行比较。在您的比较器函数中，abs(1-4) <= 3 为真，因此该函数返回假。这导致 find 函数认为 1 和 4 相等，因此它返回一个指向 1 的迭代器。

要解决这个问题，您应该确保您的比较器函数只比较元素的相对顺序，而不是它们的相等性。这可以通过在比较器函数中添加单独的相等性检查，或使用不同的容器来完成，例如 std::unordered_set，它使用 == 运算符来检查元素的相等性。