Time: 255 ms (72.29%), Space: 63.9 MB (28.89%) - LeetHub

Author: Deepak5113

0380-insert-delete-getrandom-o1/0380-insert-delete-getrandom-o1.py

@@ -0,0 +1,32 @@
+class RandomizedSet:
+
+    def __init__(self):
+        self.dict = {}
+        self.list = []
+
+    def insert(self, val: int) -> bool:
+        if val in self.dict:
+            return False
+        self.dict[val] = len(self.list)
+        self.list.append(val)
+        return True
+
+    def remove(self, val: int) -> bool:
+        if val not in self.dict:
+            return False
+        idx = self.dict[val]
+        last_element = self.list[-1]
+        self.list[idx] = last_element
+        self.dict[last_element] = idx
+        self.list.pop()
+        del self.dict[val]
+        return True
+
+    def getRandom(self) -> int:
+        return random.choice(self.list)
+
+# 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()