力扣题解:设计哈希集合

96 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第21天,点击查看活动详情

题目描述

原题链接 :

705. 设计哈希集合 - 力扣(LeetCode)

不使用任何内建的哈希表库设计一个哈希集合(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

思路分析

  1. 设计一个hashset,最重要的就是设计hash函数,和处理哈希冲突,此处采用链地址法进行哈希冲突处理;
  2. 由于0 < key <=1000000。所以此处设计hash函数为 key % 1000;
  3. 设置一个vector<vector> 来进行数据存储,index为对应的key % 1000,如果index中遇到相同的,就直接进行vec追加即可;
  4. 注意add之前需要判定一下对应的key是否已经存在了,因为set是全局唯一的,不能存在重复的key.

AC 代码

class MyHashSet {
public:
    /** Initialize your data structure here. */
    MyHashSet() {
        vector<vector<int>> tmpVec(1000, vector<int>());
        hashVec = tmpVec;
    }

    void add(int key) {
        if (FindAndDel(key, false)) {
            return;
        }
        hashVec.at(key % 1000).emplace_back(key);
        return;
    }

    void remove(int key) {
        (void)FindAndDel(key, true);
        return;
    }

    /** Returns true if this set contains the specified element */
    bool contains(int key) {
        return FindAndDel(key, false);
    }

    bool FindAndDel(int key, bool isDel) {
        vector<int>& curPosition = hashVec.at(key % 1000);
        if (curPosition.size() == 0 || curPosition.empty()) {
            return false;
        }
        for (int i = 0; i < curPosition.size(); i++) {
            if (curPosition.at(i) == key) {
                if (isDel) {
                    curPosition.erase(curPosition.begin() + i);
                }
                return true;
            }
        }
        return false;
    }
private:
    vector<vector<int>> hashVec;
};

参考

[705. 设计哈希集合]c++多种解题思路对比 - 设计哈希集合 - 力扣(LeetCode)