Leetcode 每日一题
题目链接: 381. O(1) 时间插入、删除和获取随机元素 - 允许重复
难度: 困难
解题思路: 这道题其实就是考一些语言特性,对于Python 语言我们可以直接用list来完成,插入使用append,删除使用remove,随机取数就随机一个int作为下标直接取。
题解:
import random
class RandomizedCollection:
def __init__(self):
"""
Initialize your data structure here.
"""
self.val_list = []
def insert(self, val: int) -> bool:
"""
Inserts a value to the collection. Returns true if the collection did not already contain the specified element.
"""
if val in self.val_list:
self.val_list.append(val)
return False
else:
self.val_list.append(val)
return True
def remove(self, val: int) -> bool:
"""
Removes a value from the collection. Returns true if the collection contained the specified element.
"""
if val in self.val_list:
self.val_list.remove(val)
return True
else:
return False
def getRandom(self) -> int:
"""
Get a random element from the collection.
"""
list_len = len(self.val_list)
if list_len > 0:
index = random.randint(0, list_len - 1)
return self.val_list[index]
else:
return None
# Your RandomizedCollection object will be instantiated and called as such:
# obj = RandomizedCollection()
# param_1 = obj.insert(val)
# param_2 = obj.remove(val)
# param_3 = obj.getRandom()