smj007

Untitled

Jun 19th, 2022 (edited)
171
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.77 KB | None | 0 0
  1. # More Code
  2. class Solution:
  3.     def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
  4.        
  5.         if root is None:
  6.             return root
  7.        
  8.         if not root.left and not root.right:
  9.             return root
  10.        
  11.         root.left = self.invertTree(root.left)
  12.         root.right = self.invertTree(root.right)
  13.        
  14.         tmp = root.left
  15.         root.left = root.right
  16.         root.right= tmp
  17.        
  18.         return root
  19.  
  20.  
  21. # Simpler
  22. class Solution:
  23.     def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
  24.  
  25.         if root is None:
  26.             return
  27.  
  28.         left = self.invertTree(root.right)
  29.         right = self.invertTree(root.left)
  30.         root.left, root.right = left, right
  31.  
  32.         return root
Advertisement
Add Comment
Please, Sign In to add comment