-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbtree.py
More file actions
67 lines (58 loc) · 1.59 KB
/
btree.py
File metadata and controls
67 lines (58 loc) · 1.59 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
class node:
def __init__(self, inData):
self.data = inData
self.left = None
self.right = None
class bTree:
def __init__(self):
self.root = None
def insertNode(self, currentNode, newNode):
if currentNode.data > newNode.data:
if currentNode.left is None:
currentNode.left = newNode
else:
self.insertNode(currentNode.left, newNode)
else:
if currentNode.right is None:
currentNode.right = newNode
else:
self.insertNode(currentNode.right, newNode)
def insert(self, data):
newNode = node(data)
if self.root is None:
self.root = newNode
else:
self.insertNode(self.root, newNode)
def deptFirstPrintNode(self, node):
print(node.data)
if node.left is not None:
self.deptFirstPrintNode(node.left)
if node.right is not None:
self.deptFirstPrintNode(node.right)
def depthFirstPrint(self):
self.deptFirstPrintNode(self.root)
def transverseNodesDoing(self, node, function):
function(node)
if node.left is not None:
self.transverseNodesDoing(node.left, function)
if node.right is not None:
self.transverseNodesDoing(node.right, function)
#def bfPrint(self):
#bfPrintNode(self.root)
#def bfPrintNode(self, node):
btree = bTree()
btree.insert(4)
btree.insert(2)
btree.insert(1)
btree.insert(3)
btree.insert(6)
btree.insert(5)
btree.insert(7)
print("normal print function:")
btree.depthFirstPrint()
print("lambda print function")
btree.transverseNodesDoing(btree.root, lambda node:print(node.data))
findLambda = lambda node: if node.data is None:
return False
if node.data =
print(btree.transverseNodesDoing btree.root, lambda node: if