unordered_map 是C++中的哈希表,可以在任意类型与类型之间做映射。
基本操作
1.引用头文件:#include <unordered_map>
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()
6.遍历
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