【leetcode】【图解】617. 合并二叉树

题目

难度:简单

给你两棵二叉树: root1root2

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

返回合并后的二叉树。

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

思路

  1. 本题可以通过常见的二叉树的遍历方式同时也是递归的方式 ),同时遍历两棵树,每次遍历获取相同的节点,然后根据以下三种情况求和
    1. 若遍历到的位置均不为空,则对两个节点求和,将求和的结果覆盖 root2 的这个节点
    2. 若遍历到的位置,root1 为空,则返回 root2 的值(root2 为空也返回)
    3. 若遍历到的位置,root2 为空,则返回 root1 的值(root1 为空也返回)
  2. 对代码的图解见代码实现部分

代码实现

本代码采用前序遍历的方式来遍历两棵二叉树

cpp 复制代码
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if (root1 == nullptr) return root2;
        if (root2 == nullptr) return root1;

		// 将新值覆盖在 root2 上
        root2->val = root2->val + root1->val;
        root2->left = mergeTrees(root1->left, root2->left);
        root2->right = mergeTrees(root1->right, root2->right);

        return root2;
    }

下面通过一个示意图来演示递归的效果(左树为 root1,右树为 root2,右侧方框表示函数栈)

首先,root1 和 root2 入栈,此时它们分别指向的值是 1,2,按照求和的要求,符合情况 1,于是 root2 的值更新为 3。

此时代码执行到了 root2->left = mergeTrees(root1->left, root2->left); 处,我们继续将 root1,root2 的左节点传入递归函数。

如图,我们看到出现了一个新的函数调用栈,此时的 root1 和 root2 为 3 和 1,它们分别是传入的上一层函数中的 root1->left 和 root2->left 的值。

同理计算 root2->val = root2->val + root1->val; 得到 root2 的值为 4。

此时代码执行到了 root2->left = mergeTrees(root1->left, root2->left); 处,我们继续将 root1,root2 的左节点传入递归函数。

在该层递归的代码中,因为 root2 为 null,所以触发了以下这段代码,于是此栈被销毁,并返回上一层栈中。

cpp 复制代码
        if (root2 == nullptr) {
            return root1;
        }

此时左节点已经处理完了,接着处理右节点(即红色节点的右节点),处理方式同上一步相同。

之后的步骤以此类推就可以完成。

时空复杂度分析

时间复杂度为为 O(n),n 为节点的个数,空间复杂度为 O(h),h 为树的高度,因为每一层树都要开辟一个新的栈空间

相关推荐
CoovallyAIHub17 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
NAGNIP18 小时前
Serverless 架构下的大模型框架落地实践
算法·架构
moonlifesudo18 小时前
半开区间和开区间的两个二分模版
算法
moonlifesudo18 小时前
300:最长递增子序列
算法
CoovallyAIHub1 天前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!
深度学习·算法·计算机视觉
CoovallyAIHub1 天前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
聚客AI2 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v2 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工2 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农2 天前
【React用到的一些算法】游标和栈
算法·react.js