-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbt.py
More file actions
45 lines (33 loc) · 876 Bytes
/
bt.py
File metadata and controls
45 lines (33 loc) · 876 Bytes
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
class BT(object):
def __init__(self, val):
self.val = val
self.r = None
self.l = None
def __str__(self):
return (self.l.__str__() + '<-' if self.l != None else '') + \
self.val.__str__() + ('->' + self.r.__str__() if self.r != None else '')
def insert(self, node):
#insert LEFT
if self.val > node.val:
if self.l == None:
self.l = node
else:
self.l.insert(node)
#insert RIGHT
else:
if self.r == None:
self.r = node
else:
self.r.insert(node)
def main():
import random
val = sorted(random.sample(xrange(10000), 10))
#val = ['h','b','c','d','e','f', 'i']
mid = len(val)/2
btreeRoot = BT(val[mid])
del val[mid]
print 'root', btreeRoot
for i in val:
btreeRoot.insert(BT(i))
print 'inserted %s:' % i, btreeRoot
main()