没事刷刷LetCode——617.合并二叉树

CSDN话题挑战赛第2期
参赛话题：算法题解

1.题目描述

给你两棵二叉树： root1 和 root2 。

想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。

输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出：[3,4,5,5,4,null,7]

2.题目链接

617.合并二叉树

3.思路讲解

我们需要使用dfs或者bfs搜索二叉树。搜索过程中会存在三种情况

1.一树节点为空，另一树节点不为空，此时直接赋值另一树节点。
2.两树节点都不为空，此时合并数值。
3.两树节点都为空，不做处理。

4.模板代码

返回新建树

TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
	// 如果有一个节点为空，则直接赋值另一个节点，对应情况1，3
    if (root1 == nullptr) {
        return root2;
    }
    if (root2 == nullptr) {
        return root1;
    }
    // 新建一棵树，首先处理头节点
    TreeNode* merged = new TreeNode(root1->val + root2->val);
    // 然后递归处理左右节点
    merged->left = mergeTrees(root1->left, root2->left);
    merged->right = mergeTrees(root1->right, root2->right);
    return merged;
}

在原树上进行修改

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