LeetCode 705. 设计哈希集合

题目描述

不使用任何内建的哈希表库设计一个哈希集合（HashSet）。

实现 MyHashSet 类：

void add(key) 向哈希集合中插入值 key 。
bool contains(key) 返回哈希集合中是否存在这个值 key 。
void remove(key) 将给定值 key 从哈希集合中删除。如果哈希集合中没有这个值，什么也不做。

示例：

输入：
["MyHashSet", "add", "add", "contains", "contains", "add", "contains", "remove", "contains"]
[[], [1], [2], [1], [3], [2], [2], [2], [2]]
输出：
[null, null, null, true, false, null, true, null, false]

解释：
MyHashSet myHashSet = new MyHashSet();
myHashSet.add(1);      // set = [1]
myHashSet.add(2);      // set = [1, 2]
myHashSet.contains(1); // 返回 True
myHashSet.contains(3); // 返回 False ，（未找到）
myHashSet.add(2);      // set = [1, 2]
myHashSet.contains(2); // 返回 True
myHashSet.remove(2);   // set = [1]
myHashSet.contains(2); // 返回 False ，（已移除）

提示：

$0 \le key \le 10^6$
最多调用 $10^4$ 次 add、remove 和 contains

拉链法

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

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

const int N = 20003;

class MyHashSet {
private:
    vector<int> h[N];
public:
    MyHashSet() {

    }

    int find(const vector<int>& h, int key) {
        for ( int i = 0; i < h.size(); i ++ )
            if ( h[i] == key ) return i;
        return -1;
    }
    
    void add(int key) {
        int k = key % N;
        int idx = find(h[k], key);
        if ( idx == -1 ) h[k].push_back(key);
    }
    
    void remove(int key) {
        int k = key % N;
        int idx = find(h[k], key);
        if ( idx != -1 ) h[k].erase(h[k].begin() + idx);
    }
    
    bool contains(int key) {
        int k = key % N;
        int idx = find(h[k], key);
        return idx != -1;
    }
};

/**
 * Your MyHashSet object will be instantiated and called as such:
 * MyHashSet* obj = new MyHashSet();
 * obj->add(key);
 * obj->remove(key);
 * bool param_3 = obj->contains(key);
 */