mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-23 21:11:08 +00:00
Compare commits
7 Commits
a6e8e8d427
...
669967da84
Author | SHA1 | Date | |
---|---|---|---|
|
669967da84 | ||
|
f3f32ae3ca | ||
|
7e010687c4 | ||
|
31bc93f1a2 | ||
|
f08e77f6aa | ||
|
c6f14ee860 | ||
|
e4a9f543a0 |
|
@ -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
|
||||
|
|
152
data_structures/binary_tree/morris_postorder_traversal.py
Normal file
152
data_structures/binary_tree/morris_postorder_traversal.py
Normal file
|
@ -0,0 +1,152 @@
|
|||
"""
|
||||
Problem Statement: Given a binary perform an postorder traversal using Morris Postorder
|
||||
traversal algorithm. (Iterative version of Postorder traversal of tree)
|
||||
|
||||
https://www.geeksforgeeks.org/morris-traversal-for-postorder/
|
||||
"""
|
||||
|
||||
|
||||
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_postorder_traversal()
|
||||
[2, 4, 5, 3, 7, 6, 12, 10, 9]
|
||||
|
||||
"""
|
||||
|
||||
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 _successor(self, node: TreeNode) -> TreeNode:
|
||||
"""
|
||||
Helper Function to return successor of the given node in a binary tree
|
||||
|
||||
Parameters:
|
||||
-----------
|
||||
node : A node in the binary tree.
|
||||
|
||||
Returns:
|
||||
--------
|
||||
The successor of the node passed in the parameter
|
||||
"""
|
||||
temp_node = node.right
|
||||
while temp_node and temp_node.left and temp_node.left != node:
|
||||
temp_node = temp_node.left
|
||||
assert temp_node is not None, "Successor should not be None"
|
||||
return temp_node
|
||||
|
||||
def morris_postorder_traversal(self) -> list[int]:
|
||||
"""
|
||||
Function for postorder traversal using morris postorder traversal.
|
||||
|
||||
Algorithm :
|
||||
------------
|
||||
First set current node as root node.
|
||||
|
||||
while current node is not empty
|
||||
If the current node has no right child.
|
||||
push the current node to temp.
|
||||
point the current node to its left child
|
||||
|
||||
else.
|
||||
find successor node of the current node.
|
||||
if successor left child points to current node,
|
||||
remove the link of the left child of successor node.
|
||||
point the current node to its left child.
|
||||
else,
|
||||
push the current node to temp.
|
||||
make the current node as the left child of the successor node.
|
||||
point current node to its right child.
|
||||
|
||||
Reverse the temp array to get postorder traversaol.
|
||||
|
||||
Returns:
|
||||
--------
|
||||
A list of integers representing the postorder traversal.
|
||||
|
||||
|
||||
|
||||
"""
|
||||
postorder_traversal = []
|
||||
current_node = self.root
|
||||
|
||||
while current_node:
|
||||
if current_node.right is None:
|
||||
postorder_traversal.append(current_node.value)
|
||||
current_node = current_node.left
|
||||
else:
|
||||
successor_node = self._successor(current_node)
|
||||
if successor_node.left == current_node:
|
||||
successor_node.left = current_node
|
||||
current_node = current_node.left
|
||||
else:
|
||||
postorder_traversal.append(current_node.value)
|
||||
successor_node.left = current_node
|
||||
current_node = current_node.right
|
||||
return postorder_traversal[::-1]
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import doctest
|
||||
|
||||
doctest.testmod()
|
Loading…
Reference in New Issue
Block a user