【C++】unordered_set和unordered_map的封装(哈希)

秦jh_ 2024-08-22 12:35:01 阅读 90

 🌈个人主页:秦jh_-CSDN博客

🔥 系列专栏:https://blog.csdn.net/qinjh_/category_12575764.html?spm=1001.2014.3001.5482

 

9efbcbc3d25747719da38c01b3fa9b4f.gif

​ 

目录

key和pair 

 仿函数hash

迭代器

operator++

HashTable.h

my_unordered_map.h

my_unordered_set.h


前言

💬 hello! 各位铁子们大家好哇。

             今日更新了unordered_map和unordered_set封装的相关内容

🎉 欢迎大家关注🔍点赞👍收藏⭐️留言📝

key和pair 

前面已经实现了哈希的底层,现用哈希进行封装。

unordered_set和unordered_map的封装和map、set大体思路一样。hash是底层,他并不知道传入的是k还是pair,但是上层的unordered_set和unordered_map知道。所以在hash多传入一个模板参数KeyOfT,这样再在map和set中分别实现取出key的逻辑即可。


 仿函数hash

 由于hash现在是底层,我们的仿函数不可能直接传给hash底层,所以得在unordered_set和unordered_map上传多一个模板参数,这样取模的仿函数就可以在外面传了。

迭代器

当遍历完一个桶后,准备找下一个桶时,就需要有哈希表,不然就找不到下一个桶,所以iterator需要传第二个参数:哈希表的指针。

operator++

当当前桶走完了,就要找下一个不为空的桶的第一个节点。循环结束有两种可能,一:所有桶都走完了。二:找到下一个不为空的桶。

HashTable.h

<code>#pragma once

#include<vector>

template<class K>

struct HashFunc

{

size_t operator()(const K& key)

{

return (size_t)key; //负数、指针都能转

}

};

//特化

template<>

struct HashFunc<string>

{

size_t operator()(const string& key)

{

size_t hash = 0;

for (auto ch : key)

{

hash *= 131;

hash += ch;

}

return hash;

}

};

namespace open_address

{

enum State

{

EMPTY,

EXIST,

DELETE

};

template<class K, class V>

struct HashData

{

pair<K, V> _kv;

State _state = EMPTY;

};

template<class K, class V, class Hash = HashFunc<K>>

class HashTable

{

public:

HashTable()

{

_tables.resize(10);

}

bool Insert(const pair<K, V>& kv)

{

if (Find(kv.first)) //不允许冗余

return false;

//扩容

if (_n * 10 / _tables.size() >= 7)

{

//方法一

//size_t newsize = _tables.size() * 2; //用vector的话需要手动映射

//vector<HashData<K, V>> newtables(newsize);

旧表重新计算负载到新表

//for(size_t i=0;i<_tables.size();i++)

//{ }

//方法二

HashTable<K, V, Hash> newHT;

newHT._tables.resize(_tables.size() * 2);

//旧表重新计算负载到新表

for (size_t i = 0; i < _tables.size(); i++)

{

if (_tables[i]._state == EXIST)

{

newHT.Insert(_tables[i]._kv);//用HashTable对象插入,可以复用Insert,不需要手动映射

} //newHT已经是扩容好的了,就跳过扩容,直接来到探测部分

} //新表插入好后,再跟旧表互换

_tables.swap(newHT._tables);

}

Hash hs;

size_t hashi = hs(kv.first) % _tables.size(); //不能模capacity,否则会得到比size大的数,而size后面的位置不能用[]得到

//线性探测

while (_tables[hashi]._state == EXIST)

{

++hashi;

hashi %= _tables.size(); //如果往后找找不到,就回到前面继续找

}

_tables[hashi]._kv = kv;

_tables[hashi]._state = EXIST;

++_n;

return true;

}

HashData<K, V>* Find(const K& key)

{

Hash hs;

size_t hashi = hs(key) % _tables.size();

//线性探测

while (_tables[hashi]._state != EMPTY)

{

if (_tables[hashi]._state == EXIST &&

_tables[hashi]._kv.first == key)

{

return &_tables[hashi];

}

++hashi;

hashi %= _tables.size();

}

return nullptr;

}

bool Erase(const K& key)

{

HashData<K, V>* ret = Find(key);

if (ret == nullptr)

{

return false;

}

else

{

ret->_state = DELETE;

--_n;

return true;

}

}

private:

vector<HashData<K, V>> _tables;

size_t _n = 0; //有效数据个数

};

void TestHT1()

{

int a[] = { 10001,11,55,24,19,12,31 };

HashTable<int, int> ht;

for (auto e : a)

{

ht.Insert(make_pair(e, e));

}

cout << ht.Find(55) << endl;

cout << ht.Find(31) << endl;

ht.Erase(55);

cout << ht.Find(55) << endl;

cout << ht.Find(31) << endl;

}

void TestHT2()

{

int a[] = { 10001,11,55,24,19,12,31 };

HashTable<int, int> ht;

for (auto e : a)

{

ht.Insert(make_pair(e, e));

}

ht.Insert(make_pair(32, 32));

ht.Insert(make_pair(32, 32));

}

//如果key不支持强转成整形取模,就要自己提供转换成整形的仿函数

void TestHT3()

{

HashTable<string, int> ht;

ht.Insert(make_pair("sort", 1));

ht.Insert(make_pair("left", 1));

ht.Insert(make_pair("insert", 1));

//cout << StringHashFunc()("abcd") << endl;

//cout << StringHashFunc()("aadd") << endl;

}

}

namespace hash_bucket

{

template<class T>

struct HashNode

{

T _data;

HashNode<T>* _next;

HashNode(const T& data)

:_data(data)

,_next(nullptr)

{}

};

//前置声明

//HashTable需要用到__HTIterator,__HTIterator也需要用到HashTable

//所以只能用前置声明

template<class K, class T, class KeyOfT, class Hash >

class HashTable;

//template<class K, class T,class KeyOfT, class Hash >

//struct __HTIterator

//{

//typedef HashNode<T> Node;

//typedef __HTIterator<K, T, KeyOfT, Hash> Self;

//Node* _node;

//HashTable<K, T, KeyOfT, Hash>* _pht; //当前桶为空的时候需要找下一个桶,就需要表才能找到下一个桶

//__HTIterator(Node* node,HashTable<K,T,KeyOfT,Hash>* pht)

//:_node(node)

//,_pht(pht)

//{}

//T& operator*()

//{

//return _node->_data;

//}

//T* operator->()

//{

//return &_node->_data;

//}

//Self operator++()

//{

//if (_node->_next)

//{

////当前桶没走完,找当前桶的下一个节点

//_node = _node->_next;

//}

//else

//{

////当前桶走完了,找下一个不为空的桶的第一个节点

//KeyOfT kot;

//Hash hs;

//size_t i = hs(kot(_node->_data)) % _pht->_tables.size();

//++i;

//for (; i < _pht->_tables.size(); i++)

//{

//if (_pht->_tables[i])

//break;

//}

//if (i == _pht->_tables.size())

//{

////所有桶都走完了,最后一个的下一个用nullptr标记

//_node = nullptr;

//}

//else

//{

//_node = _pht->_tables[i];

//}

//}

//return *this;

//}

//bool operator!=(const Self& s)

//{

//return _node != s._node;

//}

//};

template<class K ,class T,class KeyOfT, class Hash >

class HashTable

{

typedef HashNode<T> Node;

public:

//__HTIterator需要访问HashTable的私有,所以用友元

/*template<class K, class T, class KeyOfT, class Hash >

friend struct __HTIterator;*/

//内部类

template<class Ptr, class Ref>

struct __HTIterator

{

typedef HashNode<T> Node;

typedef __HTIterator Self;

Node* _node;

const HashTable* _pht; //当前桶为空的时候需要找下一个桶,就需要表才能找到下一个桶

__HTIterator(Node* node,const HashTable* pht)

:_node(node)

,_pht(pht)

{}

Ref operator*()

{

return _node->_data;

}

Ptr operator->()

{

return &_node->_data;

}

Self operator++()

{

if (_node->_next)

{

//当前桶没走完,找当前桶的下一个节点

_node = _node->_next;

}

else

{

//当前桶走完了,找下一个不为空的桶的第一个节点

KeyOfT kot;

Hash hs;

size_t i = hs(kot(_node->_data)) % _pht->_tables.size();

++i;

for (; i < _pht->_tables.size(); i++)

{

if (_pht->_tables[i])

break;

}

if (i == _pht->_tables.size())

{

//所有桶都走完了,最后一个的下一个用nullptr标记

_node = nullptr;

}

else

{

_node = _pht->_tables[i];

}

}

return *this;

}

bool operator!=(const Self& s)

{

return _node != s._node;

}

};

typedef __HTIterator<T*,T&> iterator;

typedef __HTIterator<const T*,const T&> const_iterator;

iterator begin()

{

for (size_t i = 0; i < _tables.size(); i++)

{

Node* cur = _tables[i];

if (cur)

{//this ->HashTable*//begin写在HashTable里面,

return iterator(cur, this);//this:返回的是哈希表的指针

}

}

return end();

}

iterator end()

{

return iterator(nullptr, this);

}

const_iterator begin()const

{

for (size_t i = 0; i < _tables.size(); i++)

{

Node* cur = _tables[i];

if (cur)

{//this ->const HashTable*//begin写在HashTable里面,

return const_iterator(cur, this);//this:返回的是哈希表的指针

}

}

return end();

}

const_iterator end()const

{

return const_iterator(nullptr, this);

}

HashTable()

{

_tables.resize(10, nullptr);

_n = 0;

}

~HashTable()

{

for (size_t i = 0; i < _tables.size(); i++)

{

Node* cur = _tables[i];

while (cur)

{

Node* next = cur->_next;

delete cur;

cur = next;

}

_tables[i] = nullptr;

}

}

pair<iterator,bool> Insert(const T& data)

{

KeyOfT kot;

iterator it = Find(kot(data));

if (it!= end())

return make_pair(it,false);

Hash hs;

//扩容

//负载因子为1时扩容

if (_n == _tables.size())

{

//HashTable<K, V> newHT;

//newHT._tables.resize(_tables.size() * 2);

旧表重新计算负载到新表

//for (size_t i = 0; i < _tables.size(); i++)

//{

//Node* cur = _tables[i];

//while(cur)

//{

//newHT.Insert(cur->_kv);

//cur = cur->_next;

//}

//}

//_tables.swap(newHT._tables);

vector<Node*> newTables(_tables.size() * 2, nullptr);

for (size_t i = 0; i < _tables.size(); i++)

{

Node* cur = _tables[i];

while(cur)

{

Node* next = cur->_next;

//头插到新表的位置

size_t hashi = hs(kot(cur->_data)) % newTables.size();

cur->_next = newTables[hashi];

newTables[hashi] = cur;

cur = next;

}

_tables[i] = nullptr;

}

_tables.swap(newTables);

}

size_t hashi = hs(kot(data)) % _tables.size();

Node* newnode = new Node(data);

//头插

newnode->_next = _tables[hashi];

_tables[hashi] = newnode;

++_n;

return make_pair(iterator(newnode,this),true);

}

iterator Find(const K& key)

{

KeyOfT kot;

Hash hs;

size_t hashi = hs(key) % _tables.size();

Node* cur = _tables[hashi];

while (cur)

{

if (kot(cur->_data) == key)

{

return iterator(cur,this);

}

cur = cur->_next;

}

return end();

}

bool Erase(const K& key)

{

KeyOfT kot;

Hash hs;

size_t hashi = hs(key) % _tables.size();

Node* prev = nullptr;

Node* cur = _tables[hashi];

while (cur)

{

if (kot(cur->_data) == key)

{

//如果删除的是第一个

if (prev == nullptr)

{

_tables[hashi] = cur->_next;

}

else

{

prev->_next = cur->_next;

}

delete cur;

return true;

}

else

{

prev = cur;

cur = cur->_next;

}

}

return false;

}

private:

vector<Node*> _tables; //指针数组

size_t _n;

//vector<list<pair<K, V>>> _tables;

};

/*void TestHT1()

{

int a[] = { 10001,11,55,24,19,12,31,4,34,44 };

HashTable<int, int> ht;

for (auto e : a)

{

ht.Insert(make_pair(e, e));

}

ht.Insert(make_pair(32, 32));

ht.Insert(make_pair(32, 32));

ht.Erase(31);

ht.Erase(11);

}

void TestHT2()

{

HashTable<string, int> ht;

ht.Insert(make_pair("sort", 1));

ht.Insert(make_pair("left", 1));

ht.Insert(make_pair("insert", 1));

}*/

}

my_unordered_map.h

#pragma once

namespace qjh

{

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 hash_bucket::HashTable<K, pair<const K,V>, MapKeyOfT, Hash >::iterator iterator;

typedef typename hash_bucket::HashTable<K, pair<const K, V>, MapKeyOfT, Hash >::const_iterator const_iterator;

iterator begin()

{

return _ht.begin();

}

iterator end()

{

return _ht.end();

}

const_iterator begin() const

{

return _ht.begin();

}

const_iterator end() const

{

return _ht.end();

}

V& operator[](const K& key)

{

pair<iterator, bool> ret = insert(make_pair(key, V()));

return ret.first->second;

}

pair<iterator, bool> insert(const pair<K, V>& key)

{

return _ht.Insert(key);

}

private:

hash_bucket::HashTable<K, pair<const K, V>, MapKeyOfT,Hash > _ht;

};

void test_unordered_map1()

{

string arr[] = { "苹果","西瓜","苹果","西瓜","苹果","苹果","西瓜",

"苹果","香蕉","苹果","香蕉","苹果","草莓","苹果","草莓" };

unordered_map<string, int> countmap;

for (auto& e : arr)

{

countmap[e]++;

}

unordered_map<string,int>::iterator it = countmap.begin();

while (it != countmap.end())

{

//it->first += 'x'; //key不能修改

it->second += 1; //value可以修改

cout << it->first << ":" << it->second << endl;

++it;

}

cout << endl;

for (auto& kv : countmap)

{

cout << kv.first << ":" << kv.second << endl;

}

cout << endl;

}

}

my_unordered_set.h

#pragma once

#include"HashTable.h"

namespace qjh

{

template<class K,class Hash = HashFunc<K>>

class unordered_set

{

struct SetKeyOfT

{

const K& operator()(const K& key)

{

return key;

}

};

public:

typedef typename hash_bucket::HashTable<K, const K, SetKeyOfT, Hash>::iterator iterator;

typedef typename hash_bucket::HashTable<K, const K, SetKeyOfT, Hash>::const_iterator const_iterator;

iterator begin()

{

return _ht.begin();

}

iterator end()

{

return _ht.end();

}

const_iterator begin() const

{

return _ht.begin();

}

const_iterator end() const

{

return _ht.end();

}

pair<iterator, bool> insert(const K& key)

{

return _ht.Insert(key);

}

iterator find(const K& key)

{

return _ht.Find(key);

}

bool erase(const K& key)

{

return _ht.Erase(key);

}

private:

hash_bucket::HashTable<K,const K, SetKeyOfT, Hash> _ht;

};

void Func(const unordered_set<int>& s)

{

unordered_set<int>::iterator it = s.begin();

while (it != s.end())

{

//*it = 1;

cout << *it << " ";

++it;

}

cout << endl;

}

void test_unordered_set()

{

unordered_set<int> s;

s.insert(31);

s.insert(11);

s.insert(5);

s.insert(15);

s.insert(25);

unordered_set<int>::iterator it = s.begin();

while (it != s.end())

{

//*it = 1;

cout << *it << " ";

++it;

}

cout << endl;

for (auto e : s)

{

cout << e << " ";

}

cout << endl;

}

}



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。