blob: 9077324ea8f5c7ebd15b35393a73b63166edb802 (
plain) (
blame)
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
#!/usr/bin/env python
import random
class RandomizedSet:
def __init__(self):
self.lst = []
self.idx_map = {}
def search(self, val):
return val in self.idx_map
def insert(self, val):
if self.search(val):
return False
self.lst.append(val)
self.idx_map[val] = len(self.lst) - 1
return True
def remove(self, val):
if not self.search(val):
return False
idx = self.idx_map[val]
self.lst[idx] = self.lst[-1]
self.idx_map[self.lst[-1]] = idx
self.lst.pop()
del self.idx_map[val]
return True
def getRandom(self):
return random.choice(self.lst)
def main():
pass
if __name__ == "__main__":
main()
|
