不使用任何内建的哈希表库设计一个哈希集合(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
  • 最多调用 10^4 次 add、remove 和 contains 。

进阶:你可以不使用内建的哈希集合库解决此问题吗?

Python 解答:

class MyHashSet:
    class Node:
        def __init__(self, val):
            self.val = val
            self.next = None

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.stack = [None for i in range(1010)]
        self.base = 1009

    def add(self, key: int) -> None:
        index = key % self.base
        if not self.stack[index]:
            self.stack[index] = self.Node(key)
        else:
            p = self.stack[index]
            if p.val == key:
                return 
            while p.next:
                p = p.next
                if p.val == key:
                    return 
            p.next = self.Node(key)

    def remove(self, key: int) -> None:
        index = key % self.base
        p = self.stack[index]
        if p:
            if p.val == key:
                self.stack[index] = p.next
            else:
                pre = p
                p = p.next
                while p:
                    if p.val == key:
                        pre.next = p.next
                        break
                    pre = p
                    p = p.next
    def contains(self, key: int) -> bool:
        """
        Returns true if this set contains the specified element
        """
        index = key % self.base
        p = self.stack[index]
        while p:
            if p.val == key:
                return True
            p = p.next
        return False

# Your MyHashSet object will be instantiated and called as such:
# obj = MyHashSet()
# obj.add(key)
# obj.remove(key)
# param_3 = obj.contains(key)
最后修改日期: 2021年5月27日

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。