Skip to content

Latest commit

 

History

History
115 lines (87 loc) · 4.77 KB

chapter2-8-chinese.md

File metadata and controls

115 lines (87 loc) · 4.77 KB

std::unordered_map中使用自定义类型

当我们使用std::unordered_map代替std::map时,对于键的选择要从另一个角度出发。std::map要求键的类型可以排序。因此,元素可以进行排序。不过,当我们使用数学中的向量作为键呢?这样一来就没有判断哪个向量大于另一个向量,比如向量(0, 1)和(1, 0)无法相比较,因为它们指向的方向不同。在std::unordered_map中这都不是问题,因为不需要对键的哈希值进行排序。对于我们来说只要为类型实现一个哈希函数和等同==操作符的实现,等同操作符的是实现是为了判断两个对象是否完全相同。本节中,我们就来实验一下这个例子。

How to do it...

本节中,我们要定义一个简单的coord数据结构,其没有默认哈希函数,所以我们必须要自行定义一个。然后我们会使用coord对象来对应一些值。

  1. 包含使用std::unordered_map所必须的头文件

    #include <iostream>
    #include <unordered_map> 
  2. 自定义数据结构,这是一个简单的数据结构,还不具备对应的哈希函数:

    struct coord {
    	int x;
    	int y;
    };
  3. 实现哈希函数是为了能让类型作为键存在,这里先实现比较操作函数:

    bool operator==(const coord &l, const coord &r)
    {
    	return l.x == r.x && l.y == r.y;
    }
  4. 为了使用STL哈希的能力,我们打开了std命名空间,并且创建了一个特化的std::hash模板。其使用using将特化类型进行别名。

    namespace std
    {
    template <>
    struct hash<coord>
    {
        using argument_type = coord;
        using result_type = size_t;
  5. 下面要重载该类型的括号表达式。我们只是为coord结构体添加数字,这是一个不太理想的哈希方式,不过这里只是展示如何去实现这个函数。一个好的散列函数会尽可能的将值均匀的分布在整个取值范围内,以减少哈希碰撞。

        result_type operator()(const argument_type &c) const
        {
            return static_cast<result_type>(c.x)
           		   + static_cast<result_type>(c.y);
        }
    };
    }
  6. 我们现在可以创建一个新的std::unordered_map实例,其能结构coord结构体作为键,并且对应任意值。例子中对std::unordered_map使用自定义的类型来说,已经很不错了。让我们基于哈希进行实例化,并填充自定义类型的map表,并打印这个map表:

    int main()
    {
        std::unordered_map<coord, int> m { 
            { {0, 0}, 1}, 
            { {0, 1}, 2},
            { {2, 1}, 3}
        };
        for (const auto & [key, value] : m) {
            std::cout << "{(" << key.x << ", " << key.y
        			 << "): " << value << "} ";
        }
        std::cout << '\n';
    }
  7. 编译运行这个例子,就能看到如下的打印信息:

    $ ./custom_type_unordered_map
    {(2, 1): 3} {(0, 1): 2} {(0, 0): 1}
    

How it works...

通常实例化一个基于哈希的map表(比如: std::unordered_map)时,我们会这样写:

std::unordered_map<key_type, value_type> my_unordered_map;

编译器为我们创建特化的std::unordered_map时,这句话背后隐藏了大量的操作。所以,让我们来看一下其完整的模板类型声明:

template<
    class Key,
    class T,
    class Hash = std::hash<Key>,
    class KeyEqual = std::equal_to<Key>,
    class Allocator = std::allocator< std::pair<const Key, T> >
> class unordered_map;

这里第一个和第二个模板类型,我么填写的是coordint。另外的三个模板类型是选填的,其会使用已有的标准模板类。这里前两个参数需要我们给定对应的类型。

对于这个例子,class Hash模板参数是最有趣的一个:当我们不显式定义任何东西时,其就指向std::hash<key_type>。STL已经具有std::hash的多种特化类型,比如std::hash<std::string>std::hash<int>std::hash<unique_ptr>等等。这些类型中可以选择最优的一种类型类解决对应的问题。

不过,STL不知道如何计算我们自定义类型coord的哈希值。所以我们要使用我们定义的类型对哈希模板进行特化。编译器会从std::hash特化列表中,找到我们所实现的类型,也就是将自定义类型作为键的类型。

如果新特化一个std::hash<coord>类型,并且将其命名成my_hash_type,我们可以使用下面的语句来实例化这个类型:

std::unordered_map<coord, value_type, my_hash_type> my_unordered_map;

这样命名就很直观,可读性好,而且编译器也能从哈希实现列表中找到与之对应的正确的类型。