1.8 面试经典150题 O(1)时间插入删除和获取随机元素

nervermore9902024-04-05 22:30

实现RandomizedSet 类：

RandomizedSet() 初始化 RandomizedSet 对象

bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true ；否则，返回 false 。

bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true ；否则，返回 false 。

int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有 相同的概率 被返回。

你必须实现类的所有函数，并满足每个函数的平均时间复杂度为 O(1) 。

python 复制代码

class RandomizedSet:

    def __init__(self):
        self.map = {}

    def hasKey(self, val: int) -> bool:
        return val in self.map

    def insert(self, val: int) -> bool:
        if self.hasKey(val): return False
        self.map.update(
            {val: val}
        )
        return True

    def remove(self, val: int) -> bool:
        if not self.hasKey(val): return False
        del self.map[val]
        return True

    def getRandom(self) -> int:
        keys = list(self.map.keys())
        index = int(random.random() * len(keys))
        return keys[index]

# Your RandomizedSet object will be instantiated and called as such:
# obj = RandomizedSet()
# param_1 = obj.insert(val)
# param_2 = obj.remove(val)
# param_3 = obj.getRandom()

考虑使用字典类型解决