-
Notifications
You must be signed in to change notification settings - Fork 0
/
controlsset.py
45 lines (33 loc) · 1.02 KB
/
controlsset.py
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
43
44
45
"""A class that implements a stupid SSet (for testing purposes)"""
import bisect
from base import BaseSet
class ControlSSet(BaseSet):
def __init__(self, iterable=[]):
self.a = []
self.add_all(iterable)
def add_all(self, iterable):
for x in iterable:
self.add(x)
def add(self, x):
i = bisect.bisect_left(self.a, x)
if i == len(self.a) or self.a[i] != x:
bisect.insort_right(self.a, x)
return True
return False
def remove(self, x):
i = bisect.bisect_left(self.a, x)
if i != len(self.a) and self.a[i] == x:
del self.a[i]
return True
return False
def find(self, x):
i = bisect.bisect_left(self.a, x)
if i != len(self.a):
return self.a[i]
return None
def size(self):
return len(self.a)
def clear(self):
self.a = list()
def __iter__(self):
return self.a.__iter__()