【宽度优先搜索 BFS】LeetCode-617. 合并二叉树

最新推荐文章于 2024-10-01 05:04:32 发布

xiaoxiawancsdn

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量336

点赞数 10

分类专栏： LeetCode每日一题文章标签：宽度优先 leetcode 算法

本文链接：https://blog.csdn.net/xiaoxiawancsdn/article/details/135122030

版权

LeetCode每日一题专栏收录该内容

63 篇文章 0 订阅

订阅专栏

617. 合并二叉树。

给你两棵二叉树： 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. 主要思路是以第一棵树为基准，将第二棵树往第一课树上添加。
1. 从根结点开始递归，递归时分为四种情况：l 和 r 均为空，则返回空；l为空，r不为空，则返回 r；l不为空，r 为空，则返回l；l和 r均不为空，则分别递归左子树和右子树，令l的左儿子指向递归返回的左子树，l的右儿子指向递归返回的右子树，最后返回 l。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode mergeTrees(TreeNode l, TreeNode r) {
        if (l == null && r == null) return null;
        if (l == null) return r;
        if (r == null) return l;
        l.val += r.val;
        l.left = mergeTrees(l.left, r.left);
        l.right = mergeTrees(l.right, r.right);
        return l; 
    }
}

复杂性分析

时间复杂度：O(n + m)
空间复杂度：O(1)