当前位置: 首页 > news >正文

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

题目

难度:简单

给你两棵二叉树: root1root2

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

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。
在这里插入图片描述

思路

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

代码实现

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

    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,所以触发了以下这段代码,于是此栈被销毁,并返回上一层栈中。

        if (root2 == nullptr) {return root1;}

在这里插入图片描述
此时左节点已经处理完了,接着处理右节点(即红色节点的右节点),处理方式同上一步相同。

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

时空复杂度分析

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

http://www.lryc.cn/news/124498.html

相关文章:

  • 基于java的汽车改装方案网站设计与实现
  • DC电源模块减小输入电源与输出负载之间的能量损失
  • Python自动化小技巧16——分类汇总写入excel不同sheet表
  • FlexRay汽车总线静电防护,如何设计保护方案图?
  • jpg图片太大怎么压缩?这样做轻松压缩图片
  • B057-spring增强 依赖注入 AOP 代理模式 创建Bean
  • 小程序多图片组合
  • YOLO v8目标跟踪详细解读(二)
  • 【广州华锐视点】AR电力职业技能培训系统让技能学习更“智慧”
  • C#学习,反射
  • 代理模式概述
  • 最新AI系统ChatGPT网站程序源码+搭建教程/公众号/H5端/安装配置教程/完整知识库
  • 前端Flex布局
  • 文盘Rust -- Mutex解决并发写文件乱序问题 | 京东云技术团队
  • 数据结构算法--2 冒泡排序,选择排序,插入排序
  • 秋招面经——快手
  • 【STM32RT-Thread零基础入门】 2. 新建RT-Thread项目
  • 别人直播的时候怎么录屏?分享一些录屏方法
  • React Native 在高IOS版本下无法显示图片的问题处理
  • SSH远程连接MacOS catalina并进行终端颜色配置
  • 用JSON.toJSONString转JSON时,属性的值为null时,输出的JSON里没有该属性
  • Java版企业电子招标采购系统源码—企业战略布局下的采购寻源tbms
  • 轻拍牛头(约数)
  • Vc - Qt - 绘制窗口背景色
  • js和cocos creator学习笔记
  • Ceph分布式存储系统
  • 阿里云SMS,APi接口返回错误码
  • Floyd算法
  • SpringBoot究竟应该如何学习?
  • 为什么很多人认为ChatGPT最好的替代工具是Claude?