phatboislym

Untitled

Jan 9th, 2024
1,316
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.77 KB | Source Code | 0 0
  1. class Solution:
  2.     def leafSimilar(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> bool:
  3.         leaf_seq1: list[int] = self.traverse_tree(root1)
  4.         leaf_seq2: list[int] = self.traverse_tree(root2)
  5.            
  6.         return leaf_seq1 == leaf_seq2
  7.  
  8.  
  9.     def traverse_tree(self, node: Optional[TreeNode]) -> list[int]:
  10.         leaves: list [int] = []
  11.         values: list[int] = []
  12.  
  13.         if not node:
  14.             return
  15.        
  16.         if not node.right and not node.left:
  17.             leaves.append(node.val)
  18.        
  19.         values = self.traverse_tree(node.left)
  20.         if values:
  21.             leaves += values
  22.        
  23.         values = self.traverse_tree(node.right)
  24.         if values:
  25.             leaves += values
  26.  
  27.         return leaves
Advertisement
Add Comment
Please, Sign In to add comment