LeetCode617 合并二叉树

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

题意

给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点。

解法

又是之前做过的一道题了
写出来了但没有用上最优解法
事实上，如果有一个节点为null，直接返回另一个节点就可以了
不需要新建一棵树，在原先的树上修改合并就可以了
递归处理左右子树

代码

class Solution {
public:
    TreeNode* merge(TreeNode* root1, TreeNode* root2) {
        if (root1 == nullptr) return root2;
        if (root2 == nullptr) return root1;
        root1 -> left = merge(root1 -> left, root2 -> left);
        root1 -> right = merge(root1 -> right, root2 -> right);
        root1 -> val += root2 -> val;
        return root1;
    }
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        return merge(root1, root2);
        
    }
};