std::map 自定义类型作为key

昨天给同事写了一个把自定义类型作为map中key值的示例，结果过了半个小时，同事反馈：不满足需求。

嗯哼？作为一个程序员，不满足需求那可就是BUG呀~ 不行，得尽快给处理一下。

【1】异常示例（不满足需求样例）

源代码如下：

#include <map>
#include <string>
#include <iostream>
using namespace std;

struct SNCloneInfo
{
    int cloneId;
    std::string type;
};

struct SNCloneCompare
{
    bool operator()(const SNCloneInfo& p1, const SNCloneInfo& p2) const
    {
        return (p1.cloneId < p2.cloneId) || (p1.cloneId == p2.cloneId && p1.type.size() < p2.type.size());
    }
};

struct SNCloneInfoRepo
{
    void create(SNCloneInfo src, SNCloneInfo dst)
    {
        m_mapRelations[src] = dst;
    }
    SNCloneInfo find(int id, std::string type)
    {
        return m_mapRelations[SNCloneInfo{ id, type }];
    }

    std::map<SNCloneInfo, SNCloneInfo, SNCloneCompare> m_mapRelations;
};

int main()
{
    SNCloneInfoRepo repo;
    repo.create(SNCloneInfo{ 0, "abc" }, SNCloneInfo{ 0, "abc" });
    repo.create(SNCloneInfo{ 0, "abcd" }, SNCloneInfo{ 1, "abc" });
    repo.create(SNCloneInfo{ 0, "bcd" }, SNCloneInfo{ 1, "abc" });
    repo.create(SNCloneInfo{ 1, "bcd" }, SNCloneInfo{ 0, "bcd" });
    repo.create(SNCloneInfo{ 2, "cde" }, SNCloneInfo{ 0, "cde" });

    cout << "repo size : " << repo.m_mapRelations.size() << endl;
    for (auto& item : repo.m_mapRelations)
    {
        cout << "K[ " << item.first.cloneId << "," << item.first.type << " ] -->"
            << "V[ " << item.second.cloneId << "," << item.second.type << " ]" << endl;
    }
    auto temp = repo.find(0, "abc");
    cout << "temp info cloneid :  " << temp.cloneId << " || type : " << temp.type << endl;

    return 0;
}

运行结果如下图：

【2】正常示例（满足需求样例）

源代码如下：

#include <map>
#include <string>
#include <iostream>
using namespace std;

struct SNCloneInfo
{
    int cloneId;
    std::string type;
};

struct SNCloneCompare 
{
    bool operator()(const SNCloneInfo& p1, const SNCloneInfo& p2) const
    {
        return (p1.cloneId < p2.cloneId) || (p1.cloneId == p2.cloneId && p1.type < p2.type);
    }
};

struct SNCloneInfoRepo
{
    void create(SNCloneInfo src, SNCloneInfo dst)
    {
        m_mapRelations[src] = dst;
    }
    SNCloneInfo find(int id, std::string type)
    {
        return m_mapRelations[SNCloneInfo{ id, type }];
    }

    std::map<SNCloneInfo, SNCloneInfo, SNCloneCompare> m_mapRelations;
};

int main()
{
    SNCloneInfoRepo repo;
    repo.create(SNCloneInfo{ 0, "abc" }, SNCloneInfo{ 0, "abc" });
    repo.create(SNCloneInfo{ 0, "abcd" }, SNCloneInfo{ 1, "abc" });
    repo.create(SNCloneInfo{ 0, "bcd" }, SNCloneInfo{ 1, "abc" });
    repo.create(SNCloneInfo{ 1, "bcd" }, SNCloneInfo{ 0, "bcd" });
    repo.create(SNCloneInfo{ 2, "cde" }, SNCloneInfo{ 0, "cde" });

    cout << "repo size : " << repo.m_mapRelations.size() << endl;
    for (auto& item : repo.m_mapRelations)
    {
        cout << "K[ " << item.first.cloneId << "," << item.first.type << " ] -->"
            << "V[ " << item.second.cloneId << "," << item.second.type << " ]" << endl;
    }
    auto temp = repo.find(0, "abc");
    cout << "temp info cloneid :  " << temp.cloneId << " || type : " << temp.type << endl;

    return 0;
}

运行结果如下图：

【3】总结

代码很简单，不做赘述。只讲差异点：

第一个示例，明明存进去了5个键值对，结果打印个数只有4个！！

经全部打印值分析，很明确错误原因：比较函数中当cloneId相同时，对字符串类型type值作了长度比较。

第一次插入的键为“0, abc”，第三次插入的键为“0, bcd”，cloneId相同，且两个键type值长度相同，map将其视作相同的键，然后把键“0，abc”的值替换为“1，abc”完毕。

经更正，详见第二个示例效果。

以此备录，警醒自己。

 

good good study, day day up.

顺序 选择 循环 总结