LeetCode617. 合并二叉树

合集 - LeetCode(37)

1.力扣151. 反转字符串中的单词2024-07-092.LeetCode515.在每个树行中找最大值2024-07-223.LeetCode104.二叉树的最大深度2024-07-224.LeetCode111.二叉树的最小深度2024-07-225.LeetCode136. 只出现一次的数字2024-07-236.LeetCode260. 只出现一次的数字 III2024-07-237.LeetCode151. 反转字符串中的单词2024-07-238.LeetCode225. 用队列实现栈2024-07-249.LeetCode102.二叉树的层序遍历2024-07-2310.LeetCode107.二叉树的层序遍历II2024-07-2311.LeetCode199.二叉树的右视图2024-07-2412.LeetCode637.二叉树的层平均值2024-07-2413.LeetCode226. 翻转二叉树2024-07-2414.LeetCode455.分发饼干2024-07-2415.LeetCode860. 柠檬水找零2024-07-2416.LeetCode122. 买卖股票的最佳时机 II2024-07-2517.LeetCode1005. K 次取反后最大化的数组和2024-07-2818.LeetCode53. 最大子数组和2024-07-2519.LeetCode513. 找树左下角的值2024-07-2520.LeetCode135. 分发糖果2024-07-2521.LeetCode222.完全二叉树的节点个数2024-07-2822.LeetCode700. 二叉搜索树中的搜索2024-07-2823.LeetCode98. 验证二叉搜索树2024-07-2924.LeetCode530. 二叉搜索树的最小绝对差2024-07-2825.LeetCode654. 最大二叉树2024-07-28

26.LeetCode617. 合并二叉树2024-07-29

27.回溯算法介绍以及模板2024-08-1428.LeetCode216.组合总和lll2024-08-1429.LeetCode39. 组合总和2024-08-1430.LeetCode40.组合总和II2024-08-1431.动态规划方法论2024-08-2032.线性dp：编辑距离2024-08-2333.线性dp：大盗阿福（打家劫舍）2024-08-2434.LeetCode300.最长递增子序列2024-08-2335.线性dp：LeetCode674. 最长连续递增序列2024-08-2436.线性dp：LeetCode516 .最长回文子序列2024-09-0737.线性dp：LeetCode122.买卖股票的最佳时机ll2024-09-10

收起

题目链接：https://leetcode.cn/problems/merge-two-binary-trees/description/

题目叙述：

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

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

返回合并后的二叉树。

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

示例 1：

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

示例 2：

输入：root1 = [1], root2 = [1,2]
输出：[2,2]

提示：

两棵树中的节点数目在范围 [0, 2000] 内
-10^4 <= Node.val <= 10^4

思路：

这题我们需要对两棵树分别进行操作，其实对操作一棵树是一模一样的，只要我们同步两棵树的操作过程就好了

上递归三部曲

1. 递归函数的参数和返回值，这题明显我们要接收两个根节点root1，root2作为参数传入，我们可以选择构造新的树来存储变化后的树，也可也选择直接在root1或root2上面进行修改。我们这里选择直接在root1上面进行修改
2. 递归函数的终止条件（递归的边界）：
   因为是传入了两个树，那么就有两个树遍历的节点t1 和 t2，如果t1 == NULL 了，两个树合并就应该是 t2 了（如果t2也为NULL也无所谓，合并之后就是NULL）。
   反过来如果t2 == NULL，那么两个数合并就是t1（如果t1也为NULL也无所谓，合并之后就是NULL）

if (t1 == NULL) return t2; // 如果t1为空，合并之后就应该是t2
if (t2 == NULL) return t1; // 如果t2为空，合并之后就应该是t1

3.单层递归的逻辑：

单层递归的逻辑比较好写，我们只需要将root2结点的值加到root1上面，然后递归构造root1的左子树和root1的右子树就行了

        root1->val+=root2->val;
        root1->left=mergeTrees(root1->left,root2->left);
        root1->right=mergeTrees(root1->right,root2->right);

代码：

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

迭代法

递归法能做的，迭代法也能做！下面是迭代法

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if (t1 == NULL) return t2;
        if (t2 == NULL) return t1;
        queue<TreeNode*> que;
        que.push(t1);
        que.push(t2);
        while(!que.empty()) {
            TreeNode* node1 = que.front(); que.pop();
            TreeNode* node2 = que.front(); que.pop();
            // 此时两个节点一定不为空，val相加
            node1->val += node2->val;
 
            // 如果两棵树左节点都不为空，加入队列
            if (node1->left != NULL && node2->left != NULL) {
                que.push(node1->left);
                que.push(node2->left);
            }
            // 如果两棵树右节点都不为空，加入队列
            if (node1->right != NULL && node2->right != NULL) {
                que.push(node1->right);
                que.push(node2->right);
            }
 
            // 当t1的左节点 为空 t2左节点不为空，就赋值过去
            if (node1->left == NULL && node2->left != NULL) {
                node1->left = node2->left;
            }
            // 当t1的右节点 为空 t2右节点不为空，就赋值过去
            if (node1->right == NULL && node2->right != NULL) {
                node1->right = node2->right;
            }
        }
        return t1;
    }
};