【力扣算法题】合并二叉树
题目介绍
给你两棵二叉树: root1 和 root2 。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-binary-trees
题解
1. 深度优先(力扣官方解法1)
首先这题的思路离不开下面的规则
- 如果两个节点都为空,那合并节点为空
- 如果一个节点为空,那合并节点的值等于另一个节点
- 如果两个都不为空,那合并节点的值等于两节点之和
运用深度优先的思路,这题的代码量会比较少,因为我们可以直接把
C++源码:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == nullptr) {
return t2;
}
if (t2 == nullptr) {
return t1;
}
auto merged = new TreeNode(t1->val + t2->val);
merged->left = mergeTrees(t1->left, t2->left);
merged->right = mergeTrees(t1->right, t2->right);
return merged;
}
// 作者: