O(1)时间插入删除和获取随机元素
实现
RandomizedSet
类:
RandomizedSet()
初始化RandomizedSet
对象bool insert(int val)
当元素val
不存在时,向集合中插入该项,并返回true
;否则,返回false
。bool remove(int val)
当元素val
存在时,从集合中移除该项,并返回true
;否则,返回false
。int getRandom()
随机返回现有集合中的一项(测试用例保证调用此方法时集合中至少存在一个元素)。每个元素应该有 相同的概率 被返回。你必须实现类的所有函数,并满足每个函数的 平均 时间复杂度为
O(1)
。
class RandomizedSet:def __init__(self):self.map = {}def hasKey(self, val: int) -> bool:return val in self.mapdef insert(self, val: int) -> bool:if self.hasKey(val): return Falseself.map.update({val: val})return Truedef remove(self, val: int) -> bool:if not self.hasKey(val): return Falsedel self.map[val]return Truedef 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()
考虑使用字典类型解决