刷题-力扣-705. 设计哈希集合

705. 设计哈希集合

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/design-hashset/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

不使用任何内建的哈希表库设计一个哈希集合（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. 根据题目描述设计hash map
2. 把除以1001后的商相等的数放在同一个vector中

代码

class MyHashSet {
private:
    vector<vector<int>> hash;
    const int cst = 1001;
public:
    /** Initialize your data structure here. */
    MyHashSet() {

    }
    
    void add(int key) {
        for (auto &vec : hash) {
            if (vec[0] / cst == key / cst) {
                vec.push_back(key);
                return;
            }
        }
        hash.push_back({key});
    }
    
    void remove(int key) {
        for (auto &vec : hash) {
            if (vec[0] / cst == key / cst) {
                for (vector<int>::iterator it = vec.begin(); it != vec.end(); it) {
                    if (*it == key) {
                        vec.erase(it);
                    } else ++it;
                }
                return;
            }
        }
    }
    
    /** Returns true if this set contains the specified element */
    bool contains(int key) {
        for (auto vec : hash) {
            if (vec[0] / cst == key / cst) {
                for (auto i : vec) {
                    if (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);
 * bool param_3 = obj->contains(key);
 */