每日一练：LeeCode-617、合并二叉树【二叉树+DFS】

news2025/1/22 12:38:32

本文是力扣LeeCode-617、合并二叉树【二叉树+DFS】学习与理解过程，本文仅做学习之用，对本题感兴趣的小伙伴可以出门左拐LeeCode。

给你两棵二叉树： 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. 确定递归函数的参数和返回值
直接使用原题目的即可

2. 确定终⽌条件

传⼊了两个树，那么就有两个树遍历的节点t1 和 t2

无论t1= =null或者t2= =null，合并后都会由不为null的一方取代
若t1 和 t2都等于null，则返回null即可

        if (root1==null)return root2;
        if (root2==null)return root1;

3. 确定单层递归的逻辑

我们重复利⽤⼀下t1这个树，t1就是合并之后树的根节点。那么单层递归中，就要把两棵树的元素加到⼀起。t1->val += t2->val；
t1 的左⼦树是：合并 t1左⼦树 t2左⼦树之后的左⼦树。
t1 的右⼦树是：合并 t1右⼦树 t2右⼦树之后的右⼦树。
最终t1就是合并之后的根节点。

        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 root1, TreeNode root2) {
        if (root1==null)return root2;	// 如果t1为空，合并之后就应该是t2
        if (root2==null)return root1; 	// 如果t2为空，合并之后就应该是t1

		// 修改了t1的数值和结构
        root1.val+=root2.val;								//中
        root1.left = mergeTrees(root1.left,root2.left);		//左
        root1.right = mergeTrees(root1.right,root2.right);	//右
        return root1;
    }
}