1

考虑以下代码:

#include <iostream>
#include <map>
#include <utility>


struct Key{
    int attr1, attr2;
    Key(int attr1, int attr2) : attr1(attr1), attr2(attr2) {}

    friend bool operator== (const Key& s1, const Key& s2);
    friend bool operator< (const Key& s1, const Key& s2);
};

bool operator== (const Key& s1, const Key& s2){
    return ((s1.attr1 == s2.attr1) && (s1.attr2 == s2.attr2));
}

bool operator< (const Key& s1, const Key& s2){
    return (s1.attr1 < s2.attr1);
}

int main(void){
    std::map<Key, int> mmap;
    mmap.insert(std::make_pair(Key(10, 10), 5));
    mmap.insert(std::make_pair(Key(10, 20), 5));
    std::cout << mmap.size() << std::endl;
}

输出是1我期望的2。这似乎是由于仅attr1operator<. 但是,如果我是正确的,则比较不必是强排序,而是弱排序就足够了。这里还有其他错误还是我必须定义operator<一个

Key1 !< Key2 && Key2 !< Key1暗示Key1 == Key2

成立吗?

4

2 回答 2

6

std::map比较元素时根本不使用operartor ==。默认情况下,在这种情况下,它使用std::lesswhich 使用您的operator <. 由于您operator <唯一的 compare attr1,并且两个对象具有相同attr1的 ,因此被认为是等效的,因此您在地图中只有一个对象。

要解决此问题,您需要检查两个成员,您可以使用该std::tie技巧制作一个为您执行此操作的元组

bool operator< (const Key& s1, const Key& s2){
    return std::tie(s1.attr1, s1.attr2) < std::tie(s2.attr1, s2.attr2);
}

另一种选择是使用 a std::unordered_map,它使用散列算法和相等运算符。使用它你只能散列attr1,但让相等运算符检查两者attr1attr2确保不添加真正的重复。

于 2020-09-14T15:52:14.697 回答
0

如果我是正确的,那么比较不必是强排序,而弱排序就足够了。

你是对的。

输出是 1,我希望它是 2。

您会错误地期望,因为根据您提供的弱排序关系,这些键被认为是相等的。

这里还有其他错误吗

不。

我必须operator<为哪个定义一个吗

Key1 !< Key2 && Key2 !< Key1 implies that Key1 == Key2

成立吗?

如果您希望操作产生一个排序,其中键被认为是相等的,当且仅当Key1 == Key2是,您确实需要定义这样的运算符。

或者您可以使用可能包含非唯一键的多映射。

于 2020-09-14T17:12:27.467 回答