66DEFAULT_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
185185if __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