我一直在学习c ++。我遇到了这个问题。
我的 set 包含一个自定义结构,其中包含两个 long int & a&湾我有一个自定义比较器结构,用于比较数字,如果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 条目。任何澄清将不胜感激。
答案 0 :(得分:6)
您的比较函数应返回某个元素是否小于另一个元素,而不是它们是否相等。 (更正式地说,它必须在集合的元素上定义"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
构造了对其参数的引用元组,然后可以按字典顺序与<
进行比较。这是一种简单,通用且快速的方法,可以为不太可比的东西的集合构建一些排序。
答案 1 :(得分:5)
您的比较功能需要满足严格/弱的订购要求。
(我实际上更喜欢使用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;
}