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 ，（已移除）

思路：该题非常简单，但可以借此熟悉一下hash表和链表以及STL的部分工具。

代码：

class MyHashSet {
private:
	vector<list<int>> data;
	static const int base = 769;
	static int hash(int key){
		return key % base;
	}
	
public:
    /** Initialize your data structure here. */
    MyHashSet() : data(base){
    }
    
    void add(int key) {
		int h = hash(key);
		for(auto it = data[h].begin() ; it != data[h].end() ; it++)
		{
			if((*it) == key)
			{
				return ;
			}
		}
		data[h].push_back(key);
    }
    
    void remove(int key) {
		int h = hash(key);
		for(auto it = data[h].begin() ; it != data[h].end() ; it++)
		{
			if((*it) == key)
			{
				data[h].erase(it);
				return ;
			}
		}
    }
    
    /** Returns true if this set contains the specified element */
    bool contains(int key) {
		int h = hash(key);
		for(auto it = data[h].begin() ; it != data[h].end() ; it++)
		{
			if((*it) == 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);
 * bool param_3 = obj->contains(key);
 */

此方法称为拉链法，我们开辟一个大小为 base 的数组，数组的每个位置是一个链表。当计算出哈希值之后，就插入到对应位置的链表当中。
哈希函数：hash(x)=x mod base
由于我们使用整数除法作为哈希函数，为了尽可能避免冲突，应当将 base 取为一个质数。在这里，我们取 base=769。