这里要对 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 函数中进行测试时,可以看到两者都可以跑起来: