力扣题目——705. 设计哈希集合

注：本文的代码实现使用的是 JS（JavaScript），为前端中想使用JS练习算法和数据结构的小伙伴提供解题思路。

描述

不使用任何内建的哈希表库设计一个哈希集合（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 <= $10^6$
• 最多调用 $10^4$ 次 add、remove 和 contains 。

---

解题思路

这种题目示例中的输入与输出就不用看了，直接看解释就很清楚了，无非是想让我们自己实现一个哈希集合。对于JS来说，已经内置了Set类，我们在不使用这个类的情况下实现add、contains、remove方法，即向集合中添加元素、判断某个元素是否已经在集合中、从集合中删除元素。我们可以使用数组来模拟实现这些功能。

我们需要将设计一个哈希函数，将输入的数值映射到一个固定范围中。常用的就是对输入key与某个固定的值BASE进行取模运算。那这个值设置为多少合适呢，自己看着办哈，哈哈哈哈。其实一般来根据入key的大小去决定，但是最好是素数，这样可以使得输入均匀地映射到0-BASE-1上，想知道为啥可以去百度。质数我取值769，当然你可以换成别的素数，会稍微影响你程序的效果。

还需要一个数组hashSet用于存放数值，这个数组应该是个二维数组，大小就是BASE的大小。这个数组中的每个元素都是一个数组，用于存放映射后值相同的入key。

var MyHashSet = function() {
	// 创建用于取模的质数
    this.BASE = 769
    // 创建大小为 BASE 的数组，每个元素初始都是空的数组[]
    this.hashSet = new Array(this.BASE).fill(0).map(()=> [])
};

哈希映射函数，将key映射到0-BASE-1上

MyHashSet.prototype.hash = function(key){
    return key % this.BASE
}

判断 key是否存在于hashSet 中

MyHashSet.prototype.contains = function(key) {
	//hash 为映射后的值
    const hash = this.hash(key)
    // 判断 hashSet中下标为 hash 的这个元素对应的数组中是否存在值为 key 的元素 
    for(let item of this.hashSet[hash])
    	// 找到，就返回 true
        if(key === item) return true
    // 遍历后没返回true，那就是没找到呗
    return false 
};

调用add方法时，一定要先判断该key是否已经在哈希表中存在，若存在，就别添加了。要时不在，直接在该key映射后对应的数组中使用push方法追加上key即可。

MyHashSet.prototype.add = function(key) {
    if(this.contains(key)) return
    const hash = this.hash(key)
    this.hashSet[hash].push(key)
};

根据key所在的hashSet[hash]中的下标，删除对应位置的key

MyHashSet.prototype.remove = function(key) {
    const hash = this.hash(key)
    // this.hashSet[hash]是hash对应的 key 的数组
    for(let index in this.hashSet[hash]){
        if(key === this.hashSet[hash][index]) {
        	// 找到要删除元素的下标，使用splice函数即可删除
            this.hashSet[hash].splice(index, 1)
            return
        }
    } 
};