C++ unordered_map用法

基本操作

引用头文件(C++11)：#include <unordered_map>
定义：unordered_map<int,int>、unordered_map<string, double>…
插入：例如将(“ABC” -> 5.45) 插入unordered_map<string, double> hash中，hash[“ABC”]=5.45
查询：hash[“ABC”]会返回5.45
判断key是否存在：hash.count(“ABC”) != 0 或 hash.find(“ABC”) != hash.end()
遍历

for (auto &item : hash)
{
    cout << item.first << ' ' << item.second << endl;
}

或

for (unordered_map<string, double>::iterator it = hash.begin(); it != hash.end(); it ++ )
{
    cout << it->first << ' ' << it->second << endl;
}

进阶操作

如果想让自定义的class作为key（unordered_map<key,value>）来使用unordered_map，需要实现：
(1) 哈希函数，需要实现一个class重载operator()，将自定义class变量映射到一个size_t类型的数。一般常用std::hash模板来实现。
(2) 判断两个自定义class类型的变量是否相等的函数，一般在自定义class里重载operator==。
示例代码：

#include <iostream>
#include <vector>
#include <unordered_map>

using namespace std;

class Myclass
{
public:
    int first;
    vector<int> second;

    // 重载等号，判断两个Myclass类型的变量是否相等
    bool operator== (const Myclass &other) const
    {
        return first == other.first && second == other.second;
    }
};

// 实现Myclass类的hash函数
namespace std
{
    template <>
    struct hash<Myclass>
    {
        size_t operator()(const Myclass &k) const
        {
            int h = k.first;
            for (auto x : k.second)
            {
                h ^= x;
            }
            return h;
        }
    };
}

int main()
{
    unordered_map<Myclass, double> S;
    Myclass a = { 2, {3, 4} };
    Myclass b = { 3, {1, 2, 3, 4} };
    S[a] = 2.5;
    S[b] = 3.123;
    cout << S[a] << ' ' << S[b] << endl;
    return 0;
}

输出:
2.5 3.123

作者：yxc
链接：https://www.acwing.com/blog/content/9/
来源：AcWing