[LeetCode] 617. Merge Two Binary Trees

Seo Seung Woo·2022년 7월 31일
0
post-thumbnail

617. Merge Two Binary Trees


❔Thinking

  • 두 tree의 각 위치 value값을 더한다
  • 더한 value값을 가지는 새로운 tree를 생성한다.

💻Solution

  1. recursive
class Solution:
    def mergeTrees(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> Optional[TreeNode]:
        if root1 and root2:
            new_tree = TreeNode(root1.val + root2.val)
            new_tree.left = self.mergeTrees(root1.left, root2.left)
            new_tree.right = self.mergeTrees(root1.right, root2.right)
            return new_tree
        else:
            return root1 or root2

🗝️keypoint

  • 해당 함수를 tree의 value를 더한 값을 return하도록 만든다
  • return A or B는 둘 다 참이면 A, 둘 다 거짓이면 B, 둘 중 하나만 참이면 참인 값을 출력한다.
profile
Code for people

0개의 댓글