持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第29天,点击查看活动详情
一、题目描述:
705. 设计哈希集合 - 力扣(LeetCode) (leetcode-cn.com)
不使用任何内建的哈希表库设计一个哈希集合(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
二、思路分析:
- 定义桶类使用链表存储元素,添加添加删除和查找接口
- 定义桶数组和数组大小,初始化数组大小为一个大的质数,并创建该长度的数组用于存储桶
- 定义哈希函数使用模运算计算桶序号
- 调用桶序号对应的桶的接口实现添加删除和查找
三、AC 代码:
class Bucket {
public:
void add(int key)
{
if (contains(key))
return;
m_elements.push_front(key);
}
void remove(int key)
{
if (!contains(key))
return;
m_elements.remove(key);
}
bool contains(int key)
{
auto iter = find(m_elements.begin(), m_elements.end(), key);
return iter != m_elements.end();
}
private:
list<int> m_elements;
};
class MyHashSet {
public:
/** Initialize your data structure here. */
MyHashSet() {
mSize = 769;
mBuckets.assign(mSize, Bucket());
}
int hash(int key) {
return key % mSize;
}
void add(int key) {
int index = hash(key);
mBuckets[index].add(key);
}
void remove(int key) {
int index = hash(key);
mBuckets[index].remove(key);
}
/** Returns true if this set contains the specified element */
bool contains(int key) {
int index = hash(key);
return mBuckets[index].contains(key);
}
private:
vector<Bucket> mBuckets;
int mSize;
};