【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 ,(已移除)
class MyHashSet {
public:
MyHashSet() {
}
void add(int key) {
if(contains(key)){
return;
}
v.push_back(key);
}
void remove(int key) {
for (auto it = v.begin(); it != v.end(); /* 不需要在这里增加it++ */) {
if (*it == key) {
it = v.erase(it); // erase返回下一个有效迭代器的位置
} else {
++it; // 只有当没有删除元素时才增加迭代器
}
}
}
bool contains(int key) {
for(int i = 0; i < this->v.size(); i++){
if(key == v[i]){
return true;
}
}
return false;
}
vector<int> v;
};
/**
* 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);
*/
复杂度分析:
时间复杂度:O(N)
空间复杂度:O(N)