Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- """
- """
- class TreeNode:
- def __init__(self, value=0, left=None, right=None):
- self.value = value
- self.left = left
- self.right = right
- def height(node):
- if node is None:
- return -1 # if using 0-based height (0 for single node, -1 for empty tree)
- left_height = height(node.left)
- right_height = height(node.right)
- return max(left_height, right_height) + 1
- # Example usage:
- # Constructing a simple binary tree
- # 1
- # / \
- # 2 3
- # / \
- # 4 5
- root = TreeNode(1)
- root.left = TreeNode(2)
- root.right = TreeNode(3)
- root.left.left = TreeNode(4)
- root.left.right = TreeNode(5)
- print("Height of the tree:", height(root)) # Output: 2
Advertisement
Add Comment
Please, Sign In to add comment