leetcode 706.设计哈希映射

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 <= key, value <= 106
最多调用 104 次 put、get 和 remove 方法

进阶：你能否不使用内置的 HashMap 库解决此问题？

题解

经典线性映射，空间换时间。

class MyHashMap {
public:
    int values[1000005];
    MyHashMap() {
        memset(values,-1,sizeof(values));
    }
    
    void put(int key, int value) {
        values[key] = value;
        return;
    }
    
    int get(int key) {
        return values[key];
    }
    
    void remove(int key) {
        values[key] = -1;
        return;
    }
};

当然也可以沿用质数上界分组的写法，但需要注意一点，哈希映射是键-值对应的。在leetcode 705设计哈希集合时，相当于把value本身做键，如果这里再沿用此种方法，就会出现哈希冲突。
所以这里我们用键值对的键做键，键值对本身做值。

class MyHashMap {
public:
    int hashBase = 1171;
    vector<list<pair<int,int> > > values;
    MyHashMap() {
        values.resize(hashBase);
    }
    
    int getIndex(int key){
        return key % hashBase;
    }
    
    void put(int key, int value) {
        int index = getIndex(key);
        for(auto i = values[index].begin() ; i != values[index].end() ; ++i){
            if((*i).first == key){
                (*i).second = value;
                return;
            }
        }
        values[index].push_back({key,value});
        return;
    }
    
    int get(int key) {
        int index = getIndex(key);
        for(auto i = values[index].begin() ; i != values[index].end() ; ++i){
            if((*i).first == key) return (*i).second;
        }
        return -1;
    }
    
    void remove(int key) {
        int index = getIndex(key);
        for(auto i = values[index].begin() ; i != values[index].end() ; ++i){
            if((*i).first == key){
                values[index].erase(i);
                return;
            }
        }
        return;
    }
};