-
Notifications
You must be signed in to change notification settings - Fork 0
/
intset.py
53 lines (44 loc) · 1.5 KB
/
intset.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
46
47
48
49
50
51
52
53
class intSet(object):
"""An intSet is a set of integers
The value is represented by a list of ints, self.vals.
Each int in the set occurs in self.vals exactly once."""
def __init__(self):
"""Create an empty set of integers"""
self.vals = []
def insert(self, e):
"""Assumes e is an integer and inserts e into self"""
if not e in self.vals:
self.vals.append(e)
def member(self, e):
"""Assumes e is an integer
Returns True if e is in self, and False otherwise"""
return e in self.vals
def remove(self, e):
"""Assumes e is an integer and removes e from self
Raises ValueError if e is not in self"""
try:
self.vals.remove(e)
except:
raise ValueError(str(e) + ' not found')
def __str__(self):
"""Returns a string representation of self"""
self.vals.sort()
return '{' + ','.join([str(e) for e in self.vals]) + '}'
def intersect(self, other):
append_return = intSet()
assert type(self) == type(other)
for i in range(0,len(other.vals)):
if other.vals[i] in self.vals:
append_return.insert(other.vals[i])
#append_return.append(other.vals[i])
#return append_return
return '{' + ','.join([str(e) for e in append_return.vals]) + '}'
s = intSet()
s.insert(3)
s.insert(4)
s.insert(5)
a = intSet()
a.insert(5)
a.insert(7)
a.insert(8)
print a.intersect(s)