LC 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 <= key <= 106
最多调用 104 次 add、remove 和 contains 。

---

算法1

开大表

把数组最大容量开到1e6+，代码略

---

算法2

哈希表 + 链插法

这样数组最大容量开到1e4+即可

Java 代码

class MyHashSet {
    final int N = 10003;

    /** Initialize your data structure here. */
    int[] h;
    int[] e;
    int[] ne;
    int idx;

    public MyHashSet() {
        h = new int [N];
        e = new int [N];
        ne = new int [N];
        idx = 0;
        for (int i = 0; i < N; i ++ ) {
            h[i] = -1;
        }
    }
    
    public void add(int key) {
        int k = (key % N + N) % N;
        if (contains(key)) return ;
        e[idx] = key;
        ne[idx] = h[k];
        h[k] = idx ++ ;
    }
    
    public void remove(int key) {
        int k = (key % N + N) % N;
        for (int i = h[k], pre = -1; i != -1; pre = i, i = ne[i])
            if (e[i] == key) {
                if (pre != -1) ne[pre] = ne[i];
                else h[k] = ne[i];
                break;
            }
    }
    
    /** Returns true if this set contains the specified element */
    public boolean contains(int key) {
        int k = (key % N + N) % N;
        for (int i = h[k]; i != -1; i = ne[i])
            if (e[i] == key)
                return true;
        return false;
    }
}

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