fix: code review changes

This commit is contained in:
kuntal0901 2024-10-05 23:41:08 +05:30
parent e667b8ab45
commit 8c3783a850

View File

@ -12,12 +12,9 @@ class TreeNode:
Attributes: Attributes:
----------- -----------
value : int value : The value stored at the node.
The value stored at the node. left : Pointer to the left child node (default is None).
left : TreeNode right : Pointer to the right child node (default is None).
Pointer to the left child node (default is None).
right : TreeNode
Pointer to the right child node (default is None).
""" """
def __init__(self, value: int) -> None: def __init__(self, value: int) -> None:
@ -30,15 +27,6 @@ class BinaryTree:
""" """
Class representing a binary tree. Class representing a binary tree.
Methods:
--------
insert(value: int) -> None:
Insert a value into the binary tree following binary search tree (BST) rules.
morris_preorder_traversal() -> List[int]:
Perform preorder traversal and return list of node values.
>>> bt = BinaryTree() >>> bt = BinaryTree()
>>> bt.insert(9) >>> bt.insert(9)
>>> bt.insert(6) >>> bt.insert(6)
@ -63,8 +51,7 @@ class BinaryTree:
Parameters: Parameters:
----------- -----------
value : int value : The value to be inserted into the binary tree.
The value to be inserted into the binary tree.
""" """
if self.root is None: if self.root is None:
self.root = TreeNode(value) self.root = TreeNode(value)
@ -77,10 +64,8 @@ class BinaryTree:
Parameters: Parameters:
----------- -----------
node : TreeNode node : The current node in the binary tree.
The current node in the binary tree. value : The value to be inserted.
value : int
The value to be inserted.
""" """
if value < node.value: if value < node.value:
if node.left is None: if node.left is None:
@ -98,13 +83,11 @@ class BinaryTree:
Parameters: Parameters:
----------- -----------
node : TreeNode node : A node in the binary tree.
A node in the binary tree.
Returns: Returns:
-------- --------
TreeNode: The predecessor of the node passed in the parameter
The predecessor of the node passed in the parameter
""" """
temp_node = node.left temp_node = node.left
while temp_node and temp_node.right and temp_node.right != node: while temp_node and temp_node.right and temp_node.right != node:
@ -137,8 +120,7 @@ class BinaryTree:
Returns: Returns:
-------- --------
List[int]: A list of integers representing the preorder traversal.
A list of integers representing the preorder traversal.