LeetCode 706. 设计哈希映射

题目描述

不使用任何内建的哈希表库设计一个哈希映射（HashMap）。

实现 MyHashMap 类：

MyHashMap() 用空映射初始化对象
void put(int key, int value) 向 HashMap 插入一个键值对 (key, value) 。如果 key 已经存在于映射中，则更新其对应的值 value 。
int get(int key) 返回特定的 key 所映射的 value ；如果映射中不包含 key 的映射，返回 -1 。
void remove(key) 如果映射中存在 key 的映射，则移除 key 和它所对应的 value 。

示例：

输入：
["MyHashMap", "put", "put", "get", "get", "put", "get", "remove", "get"]
[[], [1, 1], [2, 2], [1], [3], [2, 1], [2], [2], [2]]
输出：
[null, null, null, 1, -1, null, 1, null, -1]

解释：
MyHashMap myHashMap = new MyHashMap();
myHashMap.put(1, 1); // myHashMap 现在为 [[1,1]]
myHashMap.put(2, 2); // myHashMap 现在为 [[1,1], [2,2]]
myHashMap.get(1);    // 返回 1 ，myHashMap 现在为 [[1,1], [2,2]]
myHashMap.get(3);    // 返回 -1（未找到），myHashMap 现在为 [[1,1], [2,2]]
myHashMap.put(2, 1); // myHashMap 现在为 [[1,1], [2,1]]（更新已有的值）
myHashMap.get(2);    // 返回 1 ，myHashMap 现在为 [[1,1], [2,1]]
myHashMap.remove(2); // 删除键为 2 的数据，myHashMap 现在为 [[1,1]]
myHashMap.get(2);    // 返回 -1（未找到），myHashMap 现在为 [[1,1]]

提示：

$0 \le key, value \le 10^6$
最多调用 $10^4$ 次 put、get 和 remove 方法

拉链法

和上一题一样。只需要给vector存的类型改成pair即可，因为要存key, value。

如果有删除的功能，开放寻址法代码不好写，所以使用拉链法。

$N$ 通常是总数据量的两倍。所以这题 $N$ 应该是 $10^4*2$ ，再取离这个数最近的一个质数。使用质数作为模数出现hash冲突概率相对小点。

const int N = 20003;

class MyHashMap {
private:
    vector<pair<int, int>> h[N];

public:
    MyHashMap() {}

    int find(const vector<pair<int, int>>& h, int key) {
        for ( int i = 0; i < h.size(); i ++ )
            if ( h[i].first == key ) return i;
        return -1;
    }
    
    void put(int key, int value) {
        int t = key % N;
        int idx = find(h[t], key);
        if ( idx == -1 ) h[t].push_back({key, value});
        else h[t][idx].second = value;
    }
    
    int get(int key) {
        int t = key % N;
        int idx = find(h[t], key);
        return idx == -1 ? -1 : h[t][idx].second;
    }
    
    void remove(int key) {
        int t = key % N;
        int idx  = find(h[t], key);
        if ( idx != -1 ) h[t].erase(h[t].begin() + idx);
    }
};

/**
 * Your MyHashMap object will be instantiated and called as such:
 * MyHashMap* obj = new MyHashMap();
 * obj->put(key,value);
 * int param_2 = obj->get(key);
 * obj->remove(key);
 */