Compare commits

...

9 Commits

Author SHA1 Message Date
Kuntal Gorai
6259fe652d
Merge 8c3783a850 into f3f32ae3ca 2024-11-19 12:01:23 +05:30
pre-commit-ci[bot]
f3f32ae3ca
[pre-commit.ci] pre-commit autoupdate (#12385)
updates:
- [github.com/astral-sh/ruff-pre-commit: v0.7.3 → v0.7.4](https://github.com/astral-sh/ruff-pre-commit/compare/v0.7.3...v0.7.4)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2024-11-18 22:07:12 +01:00
kuntal0901
8c3783a850 fix: code review changes 2024-10-05 23:41:08 +05:30
kuntal0901
e667b8ab45 feat: Implemented Morris preorder traversal for Binary tree 2024-10-05 19:26:11 +05:30
kuntal0901
815b049153 feat: Implemented Morris preorder traversal for Binary tree 2024-10-05 19:23:41 +05:30
kuntal0901
31bc93f1a2 feat: Implemented Morris postorder traversal for Binary tree 2024-10-05 19:22:01 +05:30
kuntal0901
f08e77f6aa deleting unecessary files 2024-10-05 19:19:06 +05:30
kuntal0901
c6f14ee860 feat: Implemented Morris postorder traversal for Binary tree 2024-10-05 19:17:09 +05:30
kuntal0901
e4a9f543a0 feat: Implemented Morris Inorder traversal for Binary tree 2024-10-05 18:40:46 +05:30
2 changed files with 151 additions and 1 deletions

View File

@ -16,7 +16,7 @@ repos:
- id: auto-walrus
- repo: https://github.com/astral-sh/ruff-pre-commit
rev: v0.7.3
rev: v0.7.4
hooks:
- id: ruff
- id: ruff-format

View File

@ -0,0 +1,150 @@
"""
Problem Statement: Given a binary perform an preorder traversal using Morris Preorder
traversal algorithm. (Iterative version of Preorder traversal of tree)
https://www.geeksforgeeks.org/morris-traversal-for-preorder/
"""
class TreeNode:
"""
Class representing a node in a binary tree.
Attributes:
-----------
value : The value stored at the node.
left : Pointer to the left child node (default is None).
right : Pointer to the right child node (default is None).
"""
def __init__(self, value: int) -> None:
self.value = value
self.left: TreeNode | None = None
self.right: TreeNode | None = None
class BinaryTree:
"""
Class representing a binary tree.
>>> bt = BinaryTree()
>>> bt.insert(9)
>>> bt.insert(6)
>>> bt.insert(10)
>>> bt.insert(3)
>>> bt.insert(7)
>>> bt.insert(12)
>>> bt.insert(2)
>>> bt.insert(5)
>>> bt.insert(4)
>>> bt.morris_preorder_traversal()
[9, 6, 3, 2, 5, 4, 7, 10, 12]
"""
def __init__(self) -> None:
self.root: TreeNode | None = None
def insert(self, value: int) -> None:
"""
Insert a value into the binary tree.
Parameters:
-----------
value : The value to be inserted into the binary tree.
"""
if self.root is None:
self.root = TreeNode(value)
else:
self._insert_recursive(self.root, value)
def _insert_recursive(self, node: TreeNode, value: int) -> None:
"""
Helper function to insert a value recursively into the tree.
Parameters:
-----------
node : The current node in the binary tree.
value : The value to be inserted.
"""
if value < node.value:
if node.left is None:
node.left = TreeNode(value)
else:
self._insert_recursive(node.left, value)
elif node.right is None:
node.right = TreeNode(value)
else:
self._insert_recursive(node.right, value)
def _predecessor(self, node: TreeNode) -> TreeNode:
"""
Helper Function to return predecessor of the given node in a binary tree
Parameters:
-----------
node : A node in the binary tree.
Returns:
--------
The predecessor of the node passed in the parameter
"""
temp_node = node.left
while temp_node and temp_node.right and temp_node.right != node:
temp_node = temp_node.right
assert temp_node is not None, "Predecessor should not be None"
return temp_node
def morris_preorder_traversal(self) -> list[int]:
"""
Function for preorder traversal using morris preorder traversal.
Algorithm :
------------
First initialize current as root node.
while current node is not empty
If the current node has no left child.
print the current node.
point the current node to its right child
else.
find predecssor node of the current node.
if predecessor right child points to current node,
remove the link of the right child of predecessor node.
point the current node to its right child.
else,
print the current node
make the current node as the right child of the predecessor node.
point current node to its left child.
Returns:
--------
A list of integers representing the preorder traversal.
"""
preorder_traversal = []
current_node = self.root
while current_node:
if current_node.left is None:
preorder_traversal.append(current_node.value)
current_node = current_node.right
else:
predecessor_node = self._predecessor(current_node)
if predecessor_node.right == current_node:
predecessor_node.right = None
current_node = current_node.right
else:
preorder_traversal.append(current_node.value)
predecessor_node.right = current_node
current_node = current_node.left
return preorder_traversal
if __name__ == "__main__":
import doctest
doctest.testmod()