mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-18 01:00:15 +00:00
Compare commits
6 Commits
7a18217227
...
a7455a6e3b
Author | SHA1 | Date | |
---|---|---|---|
|
a7455a6e3b | ||
|
f3f32ae3ca | ||
|
727686ed7a | ||
|
f08e77f6aa | ||
|
c6f14ee860 | ||
|
e4a9f543a0 |
|
@ -16,7 +16,7 @@ repos:
|
||||||
- id: auto-walrus
|
- id: auto-walrus
|
||||||
|
|
||||||
- repo: https://github.com/astral-sh/ruff-pre-commit
|
- repo: https://github.com/astral-sh/ruff-pre-commit
|
||||||
rev: v0.7.3
|
rev: v0.7.4
|
||||||
hooks:
|
hooks:
|
||||||
- id: ruff
|
- id: ruff
|
||||||
- id: ruff-format
|
- id: ruff-format
|
||||||
|
|
150
data_structures/binary_tree/morris_inorder_traversal.py
Normal file
150
data_structures/binary_tree/morris_inorder_traversal.py
Normal file
|
@ -0,0 +1,150 @@
|
||||||
|
"""
|
||||||
|
Problem Statement: Given a binary perform an inorder traversal using Morris Inorder
|
||||||
|
traversal algorithm. (Iterative version of Inorder traversal of tree)
|
||||||
|
|
||||||
|
https://www.geeksforgeeks.org/inorder-tree-traversal-without-recursion-and-without-stack/
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
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_inorder_traversal()
|
||||||
|
[2, 3, 4, 5, 6, 7, 9, 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_inorder_traversal(self) -> list[int]:
|
||||||
|
"""
|
||||||
|
Function for inorder traversal using morris inorder traversal.
|
||||||
|
|
||||||
|
Algorithm :
|
||||||
|
------------
|
||||||
|
First set current node 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 has no right child,
|
||||||
|
make the current node as the right child of the predecessor node.
|
||||||
|
point current node to its left child.
|
||||||
|
else,
|
||||||
|
remove the link of the right child of predecessor node.
|
||||||
|
print the current node
|
||||||
|
point the current node to its right child.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
--------
|
||||||
|
A list of integers representing the inorder traversal.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
"""
|
||||||
|
inorder_traversal = []
|
||||||
|
current_node = self.root
|
||||||
|
|
||||||
|
while current_node:
|
||||||
|
if current_node.left is None:
|
||||||
|
inorder_traversal.append(current_node.value)
|
||||||
|
current_node = current_node.right
|
||||||
|
else:
|
||||||
|
predecessor_node = self._predecessor(current_node)
|
||||||
|
if predecessor_node.right is None:
|
||||||
|
predecessor_node.right = current_node
|
||||||
|
current_node = current_node.left
|
||||||
|
else:
|
||||||
|
predecessor_node.right = None
|
||||||
|
inorder_traversal.append(current_node.value)
|
||||||
|
current_node = current_node.right
|
||||||
|
return inorder_traversal
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
import doctest
|
||||||
|
|
||||||
|
doctest.testmod()
|
Loading…
Reference in New Issue
Block a user