""" 0543.1 - Diameter of Binary Tree - Solution 1 - DFS """
#####################################################################################
# Imports
#####################################################################################
from typing import Optional
#####################################################################################
# Classes
#####################################################################################
class TreeNode:
"""Tree Node Class"""
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
class Solution:
"""Solution Class"""
def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
"""Diameter of Binary Tree Function"""
self.diameter = 0
def depth(node: Optional[TreeNode]) -> int:
if not node:
return 0
left = depth(node.left)
right = depth(node.right)
# Update diameter: path through this node = left depth + right depth
self.diameter = max(self.diameter, left + right)
# Return depth of this subtree
return 1 + max(left, right)
depth(root)
return self.diameter
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
# Example 1: root = [1,2,3,4,5] -> Output: 3
root1 = TreeNode(1)
root1.left = TreeNode(2)
root1.right = TreeNode(3)
root1.left.left = TreeNode(4)
root1.left.right = TreeNode(5)
print(Solution().diameterOfBinaryTree(root1)) # 3
# Example 2: root = [1,2] -> Output: 1
root2 = TreeNode(1)
root2.left = TreeNode(2)
print(Solution().diameterOfBinaryTree(root2)) # 1
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
struct N{l:Option<Box<N>>,r:Option<Box<N>>}fn d(n:&Option<Box<N>>,a:&mut i32)->i32{if let Some(x)=n{let l=d(&x.l,a);let r=d(&x.r,a);*a=(*a).max(l+r);1+l.max(r)}else{0}}fn main(){let n=Some(Box::new(N{l:Some(Box::new(N{l:Some(Box::new(N{l:None,r:None})),r:Some(Box::new(N{l:None,r:None}))})),r:Some(Box::new(N{l:None,r:None}))}));let mut a=0;d(&n,&mut a);println!("{}",a)}
""" 0543.2 - Diameter of Binary Tree - Solution 2 - Iterative Post-Order """
#####################################################################################
# Imports
#####################################################################################
from typing import Optional
#####################################################################################
# Classes
#####################################################################################
class TreeNode:
"""Tree Node Class"""
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
class Solution:
"""Solution Class"""
def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
"""Diameter of Binary Tree Function"""
if not root:
return 0
diameter = 0
depth_map = {None: 0}
stack = [(root, False)]
while stack:
node, visited = stack.pop()
if visited:
left_depth = depth_map.get(node.left, 0)
right_depth = depth_map.get(node.right, 0)
diameter = max(diameter, left_depth + right_depth)
depth_map[node] = 1 + max(left_depth, right_depth)
else:
stack.append((node, True))
if node.right:
stack.append((node.right, False))
if node.left:
stack.append((node.left, False))
return diameter
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
# Example 1: root = [1,2,3,4,5] -> Output: 3
root1 = TreeNode(1)
root1.left = TreeNode(2)
root1.right = TreeNode(3)
root1.left.left = TreeNode(4)
root1.left.right = TreeNode(5)
print(Solution().diameterOfBinaryTree(root1)) # 3
# Example 2: root = [1,2] -> Output: 1
root2 = TreeNode(1)
root2.left = TreeNode(2)
print(Solution().diameterOfBinaryTree(root2)) # 1
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
struct N{l:Option<Box<N>>,r:Option<Box<N>>}fn d(n:&Option<Box<N>>,a:&mut i32)->i32{if let Some(x)=n{let l=d(&x.l,a);let r=d(&x.r,a);*a=(*a).max(l+r);1+l.max(r)}else{0}}fn main(){let n=Some(Box::new(N{l:Some(Box::new(N{l:Some(Box::new(N{l:None,r:None})),r:Some(Box::new(N{l:None,r:None}))})),r:Some(Box::new(N{l:None,r:None}))}));let mut a=0;d(&n,&mut a);println!("{}",a)}
