STL库 —— unordered_set与unordered

这里要对 unordered_set 与 unordered_map 进行封装，封装时使用的是上一篇中学的 HashBucket 。不仅要完成封装，同时要写入迭代器。

一、HashBucket 的修改

1.1 节点的修改 T

首先来认识一下使用 unordered_set 和 ordered_map 时的区别：

unordered_set 存储唯一的键值。你只需要传入要插入的值。

#include <unordered_set>
#include <iostream>int main() {std::unordered_set<int> mySet;mySet.insert(10);mySet.insert(20);for (const auto& elem : mySet) {std::cout << elem << " ";}return 0;
}

unordered_map 存储键值对。你需要传入键和值。

#include <unordered_map>
#include <iostream>int main() {std::unordered_map<int, std::string> myMap;myMap.insert({1, "one"});myMap.insert({2, "two"});for (const auto& pair : myMap) {std::cout << pair.first << ": " << pair.second << " ";}return 0;
}

因为 unordered_set 只存储值，而 unordered_map 存储键值对并提供键到值的映射，所以当它们底层使用同一容器进行封装时，要求该容器可以兼容这两种数据类型。

template<class T>//
struct HashNode
{T _data;//HashNode* _next;HashNode(const T& data) :_data(data), _next(nullptr)//{}
};

节点模板的修改是为了满足两者的需要，满足双方的供求。

1.2 类的修改 KeyOfT

因为 unordered_map 传入的是键值对，但是并不知道键值对的键值是什么，所以在定义类模板时，会多传入一个参数 KeyOfT ，作为键值。但肯定有很多人会疑惑，传入的本身就是键值对了， pair<K, V> 中的 K 不就是键值对的键值吗？这么做不是多此一举吗？

事实上，传入 KeyOfT 而不是直接使用键 Key 的主要原因是为了提高代码的灵活性和通用性，尤其是在存储复杂对象时。以下是一个具体的例子来说明这种情况：

示例场景

假设我们有一个存储复杂对象的哈希表，这些对象有多个属性，其中一个属性作为键。

struct Employee 
{int employee_id;std::string name;std::string department;Employee(int id, const std::string& n, const std::string& d) : employee_id(id), name(n), department(d) {}
};

`KeyOfT` 提取键的机制

我们需要一种机制来从 Employee 对象中提取 employee_id 作为键。为此，我们定义一个 KeyOfEmployee 函数对象。

struct KeyOfEmployee
{int operator()(const Employee& emp) const{return emp.employee_id;}
};

键值对

键：employee_id（例如，1）
值：Employee 对象（例如，Employee(1, "Alice", "HR")）

传入示例

bool Insert(const T& obj)
{K key = KeyOfT()(obj);size_t index = Hash()(key) % _bucket.size();_bucket[index] = new T(obj);return true;
}
HashBucket<int, Employee, KeyOfEmployee, HashFunc> hb;
hb.Insert(Employee(1, "Alice", "HR"));
hb.Insert(Employee(2, "Bob", "IT"));

由上面的案例就不难看出，传入的对象不一定是键值对，有可能是自定义的类对象，此时就只需要添加 KeyOfT 的模板，就可以在类内部找到键值。

类模板添加KeyOfT

template<class K, class T, class KeyOfT, class Hash = Hashfunc<K>>
class HashBucket
{
private:vector<Node*> _bucket;size_t _n;
};

于 unordered_set 来说，KeyOfT 的方式有些多此一举。unordered_set 本质上是一个存储唯一元素的集合，没有键值对的概念。所以这里的修改是为了将就 unordered_map 。

    //unordered_settemplate<class K>class unordered_set{struct SetKeyOfT{const K& operator()(const K& Key){return Key;}};public:private:HashBucket<K, K, SetKeyOfT> _ht;//注意传参要对应HashBucket};

    //unordered_maptemplate<class K, class V>class unordered_map{struct MapKeyOfT{const K& operator()(const pair<K, V>& kv){return kv.first;}};public:private:HashBucket<K, pair<K, V>, MapKeyOfT> _ht;//注意传参要对应HashBucket};

1.3 类的修改 HashFunc

下面先来看一下这三个类与其对应的类模板：

template<class K>
class unordered_set	
{
private:HashBucket<K, K, SetKeyOfT> _ht;};template<class K, class V>
class unordered_map
{
private:HashBucket<K, pair<K, V>, MapKeyOfT> _ht;
};template<class K, class T, class KeyOfT, class Hash = HashFunc<K>>
class HashBucket
{};

在使用时，用户是直接使用 unordered_set 与 unordered_map ，所以应该希望在 unordered_set 与 unordered_map 层有一个默认的 HashFunc ，这样用户不仅可以自定义，也可以使用默认的 HashFunc ，提高了代码的灵活性，而在 HashBucket 层，只需要按照上层的指令来即可，所以就需要把默认的 HashFunc 提前到上层。

template<class K, class Hash = HashFunc<K>>
class unordered_set
{
private:HashBucket<K, K, SetKeyOfT, Hash> _ht;
};template<class K, class V, class Hash = HashFunc<K>>
class unordered_map
{
private:HashBucket<K, pair<K, V>, MapKeyOfT, Hash> _ht;
};template<class K, class T, class KeyOfT, class Hash>
class HashBucket;
{};

二、迭代器

2.1 定义迭代器成员

首先，迭代器要知道自己的位置，这就需要定义一个节点指针，另外，当在哈希桶中使用自增直到遍历完 vector 的某一节点时，因为迭代器另一个单独的类，所以需要让迭代器直到自己所处的哈希桶的结构，才好寻找下一个存在值的节点，这就需要定义一个哈希桶的指针。

这样不仅得到了迭代器的成员，也得到了迭代器的构造函数。

template<class K, class T, class KeyOfT, class Hash>
struct __HtIterator
{typedef HashNode<T> Node;Node* _node;HashBucket<K, T, KeyOfT, Hash>* _pht;__HtIterator(Node* node, HashBucket<K, T, KeyOfT, Hash>* pht):_node(node), _pht(pht){}
};

问题1：

因为迭代器中存在了哈希桶的指针来指向哈希桶，那么当遍历哈希桶的数组时，不可避免地会使用到哈希桶的 _bucket ，但是这又是个私有成员，如何解决呢？

可以使用友元来帮助解决（省略不必要的部分）：

template<class K, class T, class KeyOfT, class Hash>
class HashBucket
{
public:template<class K, class T, class KeyOfT, class Hash>friend struct __HtIterator;//友元
};

问题2：

在迭代器中，存在了哈希桶；在哈希桶中，又用到了迭代器。那么又有一个问题，编译器访问某一个的时候，必然会访问不到另一个，这是代码顺序的问题，这个问题怎么解决呢？和函数声明类似，可以在迭代器前加上哈希桶的类声明：

template<class K, class T, class KeyOfT, class Hash>//类声明
class HashBucket;template<class K, class T, class KeyOfT, class Hash>
struct __HtIterator
{};

2.2 begin 与 end 函数

2.2.1 begin 函数

如下图， begin 返回的是哈希桶第一个存值的迭代器，所以只需要挨个遍历即可，但是返回节点的地址容易，那么哈希桶的地址怎么办呢？这可是在哈希桶类中返回自己的地址，应该怎么办？

return iterator(cur, this);

其实答案很久之前就已经学过了，this指针代表的不就是本身吗？

若没有找到，可以直接返回 end ，下面只需要对 end 继续做优化即可。

template<class K, class T, class KeyOfT, class Hash>
class HashBucket
{typedef HashNode<T> Node;
public:typedef __HtIterator<K, T, KeyOfT, Hash> iterator;iterator begin(){for (size_t i = 0; i < _bucket.size(); i++){Node* cur = _bucket[i];if (cur){return iterator(cur, this);}}return end();}
private:vector<Node*> _bucket;size_t _n;
};

2.2.2 end 函数

end 返回的是最后一个存值的节点的下一个位置，所以直接可以使用空指针来构造：

	iterator end(){return iterator(nullptr, this);}

2.3 operator重载

2.3.1 自增的重载

这里有两种情况：

1.当前桶还为遍历完，那么迭代器可以直接指向当前节点的下一个。

2.当前桶已经遍历完，那么就需要遍历整个 vector ，直到找到下一个不为空的桶。

template<class K, class T, class KeyOfT, class Hash>
struct __HtIterator
{typedef HashNode<T> Node;typedef __HtIterator<K, T, KeyOfT, Hash> Self;Node* _node;HashBucket<K, T, KeyOfT, Hash>* _pht;__HtIterator(Node* node, HashBucket<K, T, KeyOfT, Hash>* pht):_node(node), _pht(pht){}Self& operator++(){if (_node->_next){//当前桶未遍历完，取桶的下一个节点_node = _node->_next;}else{//当前桶已遍历完，找下一个不为空的桶KeyOfT kot;Hash hs;size_t i = hs(kot(_node->_data)) % _pht->_bucket.size();++i;for (; i < _pht->_bucket.size(); i++){if (_pht->_bucket[i])break;}if (i == _pht->_bucket.size()) _node = nullptr;//没找到下一个不为空的桶else _node = _pht->_bucket[i];//找到了下一个不为空的桶}return *this;}
};

2.3.2 解引用和不等于的重载

这两个比较简单，就直接放在一起上代码了：

    T& operator*(){return _node->_data;}bool operator!=(const Self& s){return _node != s._node;}

2.4 完整代码

下面就可以来测试一下迭代器了，测试之前先看一下完整的代码：

2.4.1 完整的迭代器

template<class K, class T, class KeyOfT, class Hash>//
class HashBucket;template<class K, class T, class KeyOfT, class Hash>
struct __HtIterator
{typedef HashNode<T> Node;typedef __HtIterator<K, T, KeyOfT, Hash> Self;Node* _node;HashBucket<K, T, KeyOfT, Hash>* _pht;__HtIterator(Node* node, HashBucket<K, T, KeyOfT, Hash>* pht):_node(node), _pht(pht){}Self& operator++(){if (_node->_next){//当前桶未遍历完，取桶的下一个节点_node = _node->_next;}else{//当前桶已遍历完，找下一个不为空的桶KeyOfT kot;Hash hs;size_t i = hs(kot(_node->_data)) % _pht->_bucket.size();++i;for (; i < _pht->_bucket.size(); i++){if (_pht->_bucket[i])break;}if (i == _pht->_bucket.size()) _node = nullptr;//没找到下一个不为空的桶else _node = _pht->_bucket[i];//找到了下一个不为空的桶}return *this;}T& operator*(){return _node->_data;}bool operator!=(const Self& s){return _node != s._node;}
};

2.4.2 完整的HashBucket

template<class K, class T, class KeyOfT, class Hash>//
class HashBucket
{typedef HashNode<T> Node;//
public:template<class K, class T, class KeyOfT, class Hash>friend struct __HtIterator;typedef __HtIterator<K, T, KeyOfT, Hash> iterator;iterator begin(){for (size_t i = 0; i < _bucket.size(); i++){Node* cur = _bucket[i];if (cur){return iterator(cur, this);}}return end();}iterator end(){return iterator(nullptr, this);}HashBucket(){_bucket.resize(10, nullptr);_n = 0;}~HashBucket(){for (size_t i = 0; i < _bucket.size(); i++){Node* cur = _bucket[i];while (cur){Node* next = cur->_next;delete cur;cur = next;}_bucket[i] = nullptr;}}bool Insert(const T& data){KeyOfT kot;if (Find(kot(data))) return false;//Find(kv.first)->Find(kot(data))Hash hs;if (_n == _bucket.size()){vector<Node*> newBucket(_bucket.size() * 2, nullptr);for (size_t i = 0; i < _bucket.size(); i++){Node* cur = _bucket[i];while (cur){Node* next = cur->_next;size_t index = hs(kot(cur->_data)) % newBucket.size();//cur->_next = newBucket[index];newBucket[index] = cur;cur = next;}_bucket[i] = nullptr;}_bucket.swap(newBucket);}size_t index = hs(kot(data)) % _bucket.size();Node* newnode = new Node(data);newnode->_next = _bucket[index];_bucket[index] = newnode;++_n;return true;}bool Erase(const K& Key){KeyOfT kot;Hash hs;size_t index = hs(kot(Key)) % _bucket.size();Node* cur = _bucket[index];Node* prev = nullptr;while (cur){if (kot(cur->_data) == Key){//删除的是第一个节点if (prev == nullptr){_bucket[index] = cur->_next;}else{prev->_next = cur->_next;}delete cur;return true;}else{prev = cur;cur = cur->_next;}}return false;}Node* Find(const K& Key){KeyOfT kot;if (_bucket.empty()) return nullptr;Hash hs;size_t index = hs(Key) % _bucket.size();Node* cur = _bucket[index];while (cur){if (kot(cur->_data) == Key)/**/return cur;else cur = cur->_next;}return nullptr;}private:vector<Node*> _bucket;size_t _n;
};

三、迭代器的测试

3.1 重命名

迭代器的测试其实就是在 unordered_set 与 ordered_map 中复用 HashBucket 的函数，在两个类中对迭代器进行重命名，注意一定不要错了！

//unordered_set中
typedef typename HashBucket<K, K, SetKeyOfT, Hash>::iterator iterator;
//unordered_map中
typedef typename HashBucket<K, pair<K, V>, MapKeyOfT, Hash>::iterator iterator;

此外，说明一下 typename 在这里的作用：明确指出某个标识符是一个类型，从而避免编译器将其解释为非类型名称。而且，迭代器的重命名要定义在 public 域中。

3.2 unordered_set

    template<class K, class Hash = HashFunc<K>>class unordered_set{struct SetKeyOfT{const K& operator()(const K& Key){return Key;}};public:typedef typename HashBucket<K, K, SetKeyOfT, Hash>::iterator iterator;iterator begin(){return _ht.begin();}iterator end(){return _ht.end();}bool insert(const K& Key){return _ht.Insert(Key);}private:HashBucket<K, K, SetKeyOfT, Hash> _ht;};void Test_unordered_set(){unordered_set<int> s;s.insert(31);s.insert(23);s.insert(19);s.insert(6);s.insert(22);s.insert(37);for (auto e : s){cout << e << endl;}}

3.3 unordered_map

	template<class K, class V, class Hash = HashFunc<K>>class unordered_map{struct MapKeyOfT{const K& operator()(const pair<K, V>& kv){return kv.first;}};public:typedef typename HashBucket<K, pair<K, V>, MapKeyOfT, Hash>::iterator iterator;iterator begin(){return _ht.begin();}iterator end(){return _ht.end();}bool insert(const pair<K, V>& kv){return _ht.Insert(kv);}private:HashBucket<K, pair<K, V>, MapKeyOfT, Hash> _ht;};void Test_unordered_map(){unordered_map<int, int> m;m.insert(make_pair(31, 31));m.insert(make_pair(23, 23));m.insert(make_pair(19, 19));m.insert(make_pair(6, 6));m.insert(make_pair(22, 22));m.insert(make_pair(37, 37));for (auto e : m){cout << e.first << ":" << e.second << endl;}cout << endl;}

在 main 函数中进行测试时，可以看到两者都可以跑起来：