617. Merge Two Binary Trees

[rimokem]

今回の問題

https://leetcode.com/problems/merge-two-binary-trees/

次回の問題

https://leetcode.com/problems/convert-sorted-array-to-binary-search-tree/

[liquo-rice]

入力のノードを再利用するバージョンも書いたら良いと思います。

[rimokem]

書いてみたところ、以下のようになりました。
入力ノードを再利用して再帰で書くと、かなりシンプルに書けるなと感じました。

class Solution:
    def mergeTrees(
        self, root1: Optional[TreeNode], root2: Optional[TreeNode]
    ) -> Optional[TreeNode]:
        if root1 is None:
            return root2
        if root2 is None:
            return root1

        root1.val += root2.val
        root1.left = self.mergeTrees(root1.left, root2.left)
        root1.right = self.mergeTrees(root1.right, root2.right)
        return root1