617. Merge Two Binary Trees

[seal-azarashi]

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

[Yoshiki-Iwasa]

deep copyをする関数、そんなに書くの大変じゃないと思うので書いてみても良い気がしました

[liquo-rice]

こんな感じでどうでしょうか。

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        // {merged_parent, child1, child2, is_left_child}
        stack<tuple<TreeNode*, TreeNode*, TreeNode*, bool>> st;
        TreeNode dummy;
        st.push({&dummy, root1, root2, true});
        while (!st.empty()) {
            auto [merged_parent, child1, child2, is_left_child] = st.top();
            st.pop();
            if (!child1 && !child2) {
                continue;
            }
            auto merged_child = new TreeNode();
            TreeNode *left1 = nullptr, *left2 = nullptr,
                     *right1 = nullptr, *right2 = nullptr;
            if (child1) {
                merged_child->val += child1->val;
                left1 = child1->left;
                right1 = child1->right;
            }
            if (child2) {
                merged_child->val += child2->val;
                left2 = child2->left;
                right2 = child2->right;
            }
            st.push({merged_child, left1, left2, true});
            st.push({merged_child, right1, right2, false});
            if (is_left_child) {
                merged_parent->left = merged_child;
            } else {
                merged_parent->right = merged_child;
            }
        }
        return dummy.left;
    }
};

[Yoshiki-Iwasa]

良さそうです！

[Ryotaro25]

全体的に綺麗なコードだと思いまhした。

[nittoco]

良いと思いました！

[TORUS0818]

拝見しました。良いと思います。

[oda]

ここは「使う人の気持ちになって私はこうするのが一番いいと思ったのでこうしたのだ、もしかしたらもっといい方法があるのかもしれないが、少なくとも私はそう判断したのだ」と言えるならばいいと思います。
小学校の低学年の頃に、母に「きれいな字を書けるかは能力であるから人による。しかし、誠実な字を書くことは誰にでもできる。誠実な字とは読み手に伝えようという意思を持った字である。」といわれたことを思い出します。

えーっと、もうちょっと生々しく、どのように、この mergeTrees が使われるか想像しませんか。
この木構造は、何かを表していて mergeTrees 以外の破壊的な関数が存在しないならば、木の一部を共有してもいいわけですが、そうでないならば、このライブラリーを使う人にびっくりさせると思います。

それで、どんなユースケースでこの mergeTrees は使われるんでしょうか。

[nodchip]

left と right で同じ処理が重複しているのが気になりました。