lifegpc/c-utils
1
0
Fork 0
mirror of https://github.com/lifegpc/c-utils.git synced 2026-06-06 05:08:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update hash map

Browse Source
This commit is contained in:
lifegpc
2024-07-18 15:35:55 +08:00
parent 04678ac09b
commit a8298be6ca
2 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
hash_map.h
View File

@@ -110,6 +110,29 @@ void hash_map_clear(struct hash_map<K, V>* map, bool shrink = true) {
}
}
template <typename K, typename V>
bool hash_map_delete(struct hash_map<K, V>* map, K key, V* deleted_value = nullptr) {
if (!map) return false;
size_t h = map->hash(key);
size_t loc = h % map->cap;
size_t i = 1;
while (map->map[loc] && map->map[loc]->key != key) {
loc = (h + map->probing(i++)) % map->cap;
}
if (!map->map[loc]) return false;
if (deleted_value) *deleted_value = map->map[loc]->value;
if (map->free_key) map->free_key(map->map[loc]->key);
if (!deleted_value && map->free_value) map->free_value(map->map[loc]->value);
delete map->map[loc];
map->map[loc] = nullptr;
return true;
}
template <typename K, typename V, class X>
inline bool hash_map_delete(struct hash_map<K, V>* map, X key, V* deleted_value = nullptr) {
return hash_map_delete(map, K(key), deleted_value);
}
template <typename K, typename V>
struct hash_map_entry<K, V>* hash_map_get_entry(struct hash_map<K, V>* map, K key) {
if (!map) return nullptr;
@@ -136,6 +159,7 @@ bool hash_map_get(struct hash_map<K, V>* map, K key, V& value) {
while (map->map[loc] && map->map[loc]->key != key) {
loc = (h + map->probing(i++)) % map->cap;
}
if (!map->map[loc]) return false;
value = map->map[loc]->value;
return true;
}

7
test/hash_map_test.cpp
View File

@@ -31,7 +31,14 @@ TEST(HashMapTest, HashMap) {
GTEST_ASSERT_EQ(hash_map_get_entry(map, "123")->value, 123);
int v = 0;
GTEST_ASSERT_TRUE(hash_map_get(map, "234", v));
GTEST_ASSERT_FALSE(hash_map_get(map, "333", v));
GTEST_ASSERT_FALSE(hash_map_get_entry(map, "333"));
GTEST_ASSERT_EQ(v, 234);
GTEST_ASSERT_TRUE(hash_map_insert(map, "2222", 2222));
GTEST_ASSERT_TRUE(hash_map_get_entry(map, "2222"));
GTEST_ASSERT_TRUE(hash_map_delete(map, "2222", &v));
GTEST_ASSERT_EQ(v, 2222);
GTEST_ASSERT_FALSE(hash_map_get_entry(map, "2222"));
free_hash_map(map);
}