HTMLify

LeetCode - Insert Delete GetRandom O(1) - Duplicates allowed - Python
Views: 106 | Author: abh
from random import choice
class RandomizedCollection:

    def __init__(self):
        self.values = []

    def insert(self, val: int) -> bool:
        exist = not val in self.values
        self.values.append(val)
        return exist

    def remove(self, val: int) -> bool:
        exist = val in self.values
        if val in self.values:
            self.values.remove(val)
        return exist

    def getRandom(self) -> int:
        return choice(self.values)


# 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()

Comments