- Notifications
You must be signed in to change notification settings - Fork 638
Expand file tree
/
Copy pathP62_BinaryTree.py
More file actions
Latest commit
82 lines (58 loc) · 1.88 KB
/
Copy pathP62_BinaryTree.py
File metadata and controls
82 lines (58 loc) · 1.88 KB
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
# Author: OMKAR PATHAK
# A data structure in which a record is linked to two successor records, usually referred to as
# the left branch when greater and the right when less than the previous record.
classBinaryTree(object):
def__init__(self,nodeData):
self.left=None
self.right=None
self.nodeData=nodeData
defgetLeftChild(self):
returnself.left
defgetRightChild(self):
returnself.right
defsetnodeDataValue(self,value):
self.nodeData=value
defgetnodeDataValue(self):
returnself.nodeData
definsertRight(self,newnodeData):
ifself.right==None:
self.right=BinaryTree(newnodeData)
else:
tree=BinaryTree(newnodeData)
tree.right=self.right
self.right=tree
definsertLeft(self,newnodeData):
ifself.left==None:
self.left=BinaryTree(newnodeData)
else:
tree=BinaryTree(newnodeData)
self.left=tree
tree.left=self.left
defprintTree(tree):
iftree!=None:
printTree(tree.getLeftChild())
print(tree.getnodeDataValue())
printTree(tree.getRightChild())
defpprint(head_node, _pre="", _last=True, term=False):
data="*"ifhead_nodeisNoneelsehead_node.nodeData
print(_pre, "`- "if_lastelse"|- ", data, sep="")
_pre+=" "if_lastelse"| "
ifterm: return
left=head_node.getLeftChild()
right=head_node.getRightChild()
fori, childinenumerate([left, right]):
pprint(child, _pre, bool(i) ,term=not(bool(child)))
deftestTree():
myTree=BinaryTree("1")
myTree.insertLeft("2")
myTree.insertRight("3")
myTree.insertRight("4")
printTree(myTree)
pprint(myTree)
if__name__=='__main__':
testTree()
# OUTPUT
# 2
# 1
# 4
# 3