LeetCode617.合并二叉树

最新推荐文章于 2024-09-11 20:39:26 发布

快乐早睡

最新推荐文章于 2024-09-11 20:39:26 发布

阅读量1k

点赞数 29

分类专栏： LeetCode 文章标签： C++ 数据结构算法二叉树 leetcode 递归算法

本文链接：https://blog.csdn.net/weixin_40909461/article/details/138417713

版权

LeetCode 专栏收录该内容

132 篇文章 0 订阅

订阅专栏

这篇文章介绍了如何使用递归方法解决LeetCode题目617，即合并两个给定的二叉树，当节点重叠时节点值相加，否则保留非空节点。C++代码展示了合并操作的实现过程。

摘要由CSDN通过智能技术生成

在这里插入图片描述

LeetCode617.合并二叉树

题目：

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

示例：

在这里插入图片描述

输入： $roo t 1 = [1, 3, 2, 5], roo t 2 = [2, 1, 3, n u ll, 4, n u ll, 7]$
输出： $[3, 4, 5, 5, 4, n u ll, 7]$

解题思路：(递归)

确定递归函数的参数和返回值：
由于是两个树的合并，因此传入的参数为两个树的根节点，返回合并后的二叉树的根节点。
```
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2)
```
确定终止条件：
如果root1为NULL则直接返回root2，若root2为NULL则直接返回root1。
```
if(root1 == NULL) return root2;
if(root2 == NULL) return root1;
```
确定单层循环逻辑：
若两棵树的节点都不空，则将两树节点值相加：
```
root1->val += root2->val;
```
然后递归两棵树，分别从左侧和右侧来递归：
```
root1->left = mergeTrees(root1->left, root2->left);
root1->right = mergeTrees(root1->right, root2->right);
```

C++版整体代码

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1 == NULL) return root2;
        if(root2 == NULL) return root1;
        root1->val += root2->val;
        root1->left = mergeTrees(root1->left, root2->left);
        root1->right = mergeTrees(root1->right, root2->right);
        return root1;
    }
};