[LeetCode] 617. Merge Two Binary Trees

🔗

풀이

두 개의 트리를 루트 노드부터 리프 노드까지 탐색한다. 각각의 과정에서 새로운 노드들을 생성하고, 새롭게 생성된 노드의 값은 같은 위치에 있는 두 트리의 노드 값의 합을, 왼쪽과 오른쪽 자식들에 대해서는 두 트리의 노드들의 왼쪽, 오른쪽 자식의 값을 넘겨주는 방식으로 진행된다.

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

새로운 노드 인스턴스를 생성하고 val 값만 넘겨준다.