Skip to content

Commit e358a71

Browse files
committed
Renaming Node class as BinaryNode
1 parent 1825f4f commit e358a71

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

vicutils/printBin.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
DEFAULT_FILL_CHAR = " "
77

88

9-
class Node:
9+
class BinaryNode:
1010
"""
1111
Represents a node in a binary tree.
1212
@@ -46,7 +46,7 @@ def center(val, unitSize=None, fillChar=None):
4646
return str(val).center(unitSize, fillChar)
4747

4848

49-
def getDepth(node: Node):
49+
def getDepth(node: BinaryNode):
5050
"""
5151
Calculates the depth (height) of a binary tree.
5252
@@ -61,7 +61,7 @@ def getDepth(node: Node):
6161
return 1 + max(getDepth(node.left), getDepth(node.right))
6262

6363

64-
def register(node: Node, fillChar=None, unitSize=None, code="", mem=None):
64+
def register(node: BinaryNode, fillChar=None, unitSize=None, code="", mem=None):
6565
"""
6666
Recursively registers all nodes in a tree with their binary path codes.
6767
@@ -86,7 +86,7 @@ def register(node: Node, fillChar=None, unitSize=None, code="", mem=None):
8686
return mem
8787

8888

89-
def nodeToMat(node: Node, depth=-1, fillChar=None, unitSize=None, removeEmpty=True):
89+
def nodeToMat(node: BinaryNode, depth=-1, fillChar=None, unitSize=None, removeEmpty=True):
9090
"""
9191
Converts a binary tree into a 2D matrix representation for visualization.
9292
@@ -163,7 +163,7 @@ def nodeToMat(node: Node, depth=-1, fillChar=None, unitSize=None, removeEmpty=Tr
163163
return mat
164164

165165

166-
def nodeToString(node: Node, depth=-1, fillChar=None, unitSize=None, removeEmpty=True):
166+
def nodeToString(node: BinaryNode, depth=-1, fillChar=None, unitSize=None, removeEmpty=True):
167167
"""
168168
Converts a binary tree into a string representation for visualization.
169169
@@ -183,13 +183,13 @@ def nodeToString(node: Node, depth=-1, fillChar=None, unitSize=None, removeEmpty
183183

184184
# Example usage
185185
if __name__ == "__main__":
186-
myNode = Node(1)
186+
myNode = BinaryNode(1)
187187
# myNode.left = Node(2)
188-
myNode.right = Node(3)
188+
myNode.right = BinaryNode(3)
189189
# myNode.left.left = Node(4)
190190
# myNode.left.right = Node(5)
191-
myNode.right.left = Node(6)
192-
myNode.right.right = Node(7)
191+
myNode.right.left = BinaryNode(6)
192+
myNode.right.right = BinaryNode(7)
193193

194194
mat = nodeToMat(myNode)
195195
pprint(mat)

0 commit comments

Comments
 (0)