617. 合并二叉树

最新推荐文章于 2025-02-04 16:46:43 发布

我学不动了呀

最新推荐文章于 2025-02-04 16:46:43 发布

阅读量68

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode 动态规划散列表

本文链接：https://blog.youkuaiyun.com/yigeyuyu/article/details/125067283

这篇博客介绍了如何合并两棵二叉树，给出了两种不同的实现方式：一种是在原有二叉树上修改，另一种是创建新的二叉树。在合并过程中，如果节点重叠则相加其值，否则保留非null节点的值。示例代码展示了如何递归地合并左右子树，以构建合并后的二叉树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

617. 合并二叉树https://leetcode.cn/problems/merge-two-binary-trees/

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

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

注意: 合并过程必须从两个树的根节点开始。
输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出：[3,4,5,5,4,null,7]

具体有两种情况：

如果两棵树对应位置上都有节点，则新节点的值为两个节点的值相加。
如果两棵树对应位置上只有一个节点有值，则新节点的值就为该节点的值

我们可以在原有二叉树上修改，并返回修改后的二叉树，我们以返回root1为例

class Solution {
    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if(root1 == null && root2 == null){
            return null;
        }
        if(root1 == null){
            return root2;
        }
        if(root2 == null){
            return root1;
        }

        // 先合并根节点
        root1.val =root1.val+ root2.val;
        
        // 再递归合并左右子树
        root1.left = mergeTrees(root1.left, root2.left);
        root1.right = mergeTrees(root1.right, root2.right);
        return root1;
    }
}

下面我们来讨论，不修改原有二叉树的情况，那么就需要我们新建一个root

class Solution {
    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if(root1 == null && root2 == null){
            return null;
        }
        TreeNode root = new TreeNode((root1 == null ? 0 :root1.val) 
                                   + (root2 == null? 0 : root2.val));
        root.left = mergeTrees(root1 == null ? null : root1.left, root2 == null ? null: root2.left);
        root.right = mergeTrees(root1 == null ? null : root1.right, root2 == null ? null : root2.right);
        return root;
    }
}