111 lines
2.3 KiB
C++
111 lines
2.3 KiB
C++
/**
|
|
* File: array_hash_map.cpp
|
|
* Created Time: 2022-12-14
|
|
* Author: msk397 (machangxinq@gmail.com)
|
|
*/
|
|
|
|
#include "../utils/common.hpp"
|
|
|
|
/* 键值对 */
|
|
struct Pair {
|
|
public:
|
|
int key;
|
|
string val;
|
|
Pair(int key, string val) {
|
|
this->key = key;
|
|
this->val = val;
|
|
}
|
|
};
|
|
|
|
/* 基于数组实现的哈希表 */
|
|
class ArrayHashMap {
|
|
private:
|
|
vector<Pair *> buckets;
|
|
|
|
public:
|
|
ArrayHashMap() {
|
|
// 初始化数组，包含 100 个桶
|
|
buckets = vector<Pair *>(100);
|
|
}
|
|
|
|
~ArrayHashMap() {
|
|
// 释放内存
|
|
for (const auto &bucket : buckets) {
|
|
delete bucket;
|
|
}
|
|
buckets.clear();
|
|
}
|
|
|
|
/* 哈希函数 */
|
|
int hashFunc(int key) {
|
|
int index = key % 100;
|
|
return index;
|
|
}
|
|
|
|
/* 查询操作 */
|
|
string get(int key) {
|
|
int index = hashFunc(key);
|
|
Pair *pair = buckets[index];
|
|
if (pair == nullptr)
|
|
return "";
|
|
return pair->val;
|
|
}
|
|
|
|
/* 添加操作 */
|
|
void put(int key, string val) {
|
|
Pair *pair = new Pair(key, val);
|
|
int index = hashFunc(key);
|
|
buckets[index] = pair;
|
|
}
|
|
|
|
/* 删除操作 */
|
|
void remove(int key) {
|
|
int index = hashFunc(key);
|
|
// 释放内存并置为 nullptr
|
|
delete buckets[index];
|
|
buckets[index] = nullptr;
|
|
}
|
|
|
|
/* 获取所有键值对 */
|
|
vector<Pair *> pairSet() {
|
|
vector<Pair *> pairSet;
|
|
for (Pair *pair : buckets) {
|
|
if (pair != nullptr) {
|
|
pairSet.push_back(pair);
|
|
}
|
|
}
|
|
return pairSet;
|
|
}
|
|
|
|
/* 获取所有键 */
|
|
vector<int> keySet() {
|
|
vector<int> keySet;
|
|
for (Pair *pair : buckets) {
|
|
if (pair != nullptr) {
|
|
keySet.push_back(pair->key);
|
|
}
|
|
}
|
|
return keySet;
|
|
}
|
|
|
|
/* 获取所有值 */
|
|
vector<string> valueSet() {
|
|
vector<string> valueSet;
|
|
for (Pair *pair : buckets) {
|
|
if (pair != nullptr) {
|
|
valueSet.push_back(pair->val);
|
|
}
|
|
}
|
|
return valueSet;
|
|
}
|
|
|
|
/* 打印哈希表 */
|
|
void print() {
|
|
for (Pair *kv : pairSet()) {
|
|
cout << kv->key << " -> " << kv->val << endl;
|
|
}
|
|
}
|
|
};
|
|
|
|
// 测试样例请见 array_hash_map_test.cpp
|