13. unordered_map(哈希表)

基本操作

1. 引用头文件(C++11)：#include
2. 定义：unordered_map<int,int>、unordered_map<string, double> ...
3. 插入：例如将(“ABC” -> 5.45) 插入unordered_map<string, double> hash中，hash[“ABC”]=5.45
4. 查询：hash[“ABC”]会返回5.45
5. 判断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函数
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> hash;
    Myclass a = { 2, {3, 4} };
    Myclass b = { 3, {1, 2, 3, 4} };
    hash[a] = 2.5;
    hash[b] = 3.123;
    cout << hash[a] << ' ' << hash[b] << endl;
    return 0;
}

输出:

2.5 3.123

作者：yxc。著作权归作者所有