617. Merge Two Binary Trees

[TORUS0818]

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

[kazukiii]

全体的に良さそうに思います。

[kazukiii]

あった方がわかりやすいかもですが、一応この2行下で吸収できますね。

[TORUS0818]

有難うございます。思考から漏れてました。

[kazukiii]

好みの問題ですが、if not node2: でも良さそうです。

[TORUS0818]

ここは冗長ですが、二つのnodeの状態がパッとみて分かるように敢えてこうしました。

[fhiyo]

良いと思います！

[fhiyo]

好みだと思いますが、 if not (root1 and root2) の方が個人的には分かりやすいと思いました。
root1とroot2の両方があるわけじゃなければ、root1のコピーかroot2のコピーを返す。

[TORUS0818]

ドモルガンですね。私も実はこっちを先に考えて反転させました（コメントを見ていると()をあまり使わない方が好まれるような気がしたので）

[oda]

私は or 派ですね。趣味の範囲では。

[hayashi-ay]

良いと思います。

[hayashi-ay]

if文を2つに分けたほうが読みやすいと思います。

[TORUS0818]

有難うございます。
ここは過去ログを見ると、どちらを採用するか派閥が割れてそうですね（読みやすさの観点ではご指摘の通りだと思います）

[oda]

これ、
return deepcopy(root2) or deepcopy(root1)
が分かりにくくしている気がしますね。

[SuperHotDogCat]

良いと思います。if文をどう書くかは他のlogを見る限り好みの範疇なのかなと思いました

[oda]

ここで再帰していたらあまり意味がないのでは?

[oda]

スタックを積んでいる意味がないということです。
ノードを作る部分を親に押し付ける、みたいなことをする必要があります。

[TORUS0818]

確かにです、、ちょっと考えてみます。

[TORUS0818]

書き直してみました。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def mergeTrees(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root1 and not root2:
            return None
        
        merged_root = TreeNode()
        next_nodes = [(root1, root2, merged_root)]
        while next_nodes:
            node1, node2, merged_node = next_nodes.pop()
            if not node1 and not node2:
                continue
            
            if not node1 and node2:
                merged_node.val = node2.val
                if node2.right:
                    merged_node.right = TreeNode(None)
                    next_nodes.append((None, node2.right, merged_node.right))
                if node2.left:
                    merged_node.left = TreeNode(None)
                    next_nodes.append((None, node2.left, merged_node.left))
                continue

            if node1 and not node2:
                merged_node.val = node1.val
                if node1.right:
                    merged_node.right = TreeNode(None)
                    next_nodes.append((node1.right, None, merged_node.right))
                if node1.left:
                    merged_node.left = TreeNode(None)
                    next_nodes.append((node1.left, None, merged_node.left))
                continue

            merged_node.val = node1.val + node2.val
            if node1.right or node2.right:
                merged_node.right = TreeNode(None)
                next_nodes.append((node1.right, node2.right, merged_node.right))
            if node1.left or node2.left:
                merged_node.left = TreeNode(None)
                next_nodes.append((node1.left, node2.left, merged_node.left))

        return merged_root