Create random_set.py

sets/random_set.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# author: bt3gl
+
+import random
+
+class RandomizedSet:
+
+    def __init__(self):
+        self.set = []
+        self.dict = {}
+        
+    def insert(self, val: int) -> bool:
+        if val in self.dict:
+            return False
+        self.set.append(val)
+        self.dict[val] = len(self.set)
+        return True
+
+    def remove(self, val: int) -> bool:
+        if val in self.dict:
+            index = self.dict[val]
+            self.set[index] = self.set[-1]
+            self.set.pop()
+            del self.dict[val]
+            return True
+        return False
+        
+    def get_random(self) -> int:
+        return random.choice(self.set)
+