Dashboard Temp Share Shortlinks Frames API

HTMLify

LeetCode - Insert Delete GetRandom O(1) - Python
Views: 508 | Author: abh
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from random import choice
class RandomizedSet:

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

    def insert(self, val: int) -> bool:
        if val in self.values:
            return False
        self.values.append(val)
        return True
        
    def remove(self, val: int) -> bool:
        if val in self.values:
            self.values.remove(val)
            return True
        return False

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


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