Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class RandomizedSet:
- def __init__(self):
- self.d = {}
- self.q = []
- def insert(self, val: int) -> bool:
- if val in self.d:
- return False
- self.d[val] = len(self.q)
- self.q.append(val)
- return True
- def remove(self, val: int) -> bool:
- if val not in self.d:
- return False
- i = self.d[val]
- self.d[self.q[-1]] = i
- self.q[i] = self.q[-1]
- self.q.pop()
- self.d.pop(val)
- return True
- def getRandom(self) -> int:
- return choice(self.q)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement