C++如何实现自定义的struct作为unordered_set、unordered_map的key?

有两种方法:
1、自定义新struct的hash函数,和==操作符,使用的时候需要指定unordered_set的第三个模板参数:

#include <iostream>
#include <unordered_map>

struct MyStruct {
    std::string name;
    int id;

    // 自定义哈希函数
    struct Hash {
        std::size_t operator()(const MyStruct& s) const {
            // 结合哈希函数
            return std::hash<std::string>{}(s.name) ^ (std::hash<int>{}(s.id) << 1);
        }
    };

    // 自定义相等函数,用于处理哈希冲突
    bool operator==(const MyStruct& other) const {
        return name == other.name && id == other.id;
    }
};

int main() {
    std::unordered_map<MyStruct, int, MyStruct::Hash> myMap;

    // 添加元素
    MyStruct key1{"Alice", 1};
    myMap[key1] = 42;

    // 查询元素
    auto it = myMap.find(key1);
    if (it != myMap.end()) {
        std::cout << "Found: " << it->second << std::endl;
    } else {
        std::cout << "Not Found" << std::endl;
    }

    return 0;
}

2、特化std命名空间下的hash模板类。这个有个好处是不需要指定unordered_set、unordered_map第三个模板参数

#include <iostream>
#include <unordered_map>

struct MyStruct {
    std::string name;
    int id;
};

namespace std {
    template <>
    struct hash<MyStruct> {
        std::size_t operator()(const MyStruct& s) const {
            // 使用组合哈希值的方式
            return hash<std::string>{}(s.name) ^ (hash<int>{}(s.id) << 1);
        }
    };
}

int main() {
    std::unordered_map<MyStruct, int> myMap;

    // 添加元素
    MyStruct key1{"Alice", 1};
    myMap[key1] = 42;

    // 查询元素
    auto it = myMap.find(key1);
    if (it != myMap.end()) {
        std::cout << "Found: " << it->second << std::endl;
    } else {
        std::cout << "Not Found" << std::endl;
    }

    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值