1. 描述
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
2. 示例
输入:
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/ \
4 5
/ \ \
5 4 7
3. 分析
二叉树的定义是递归进行的,所以对它的操作也是递归进行的。
在示例中,Tree 1的根节点 “1” 和 Tree 2 的根节点 “2” 合并后,得到合并树的根节点 “3”, 也就是 Tree 1的根 “1” 加上 Tree 2 的根 “2”。
然后处理 Tree 1 的根节点 “1” 的左子树和 Tree 2 的根节点 “2” 的左子树;
最后处理 Tree 1 的根节点 “1” 的右子树和 Tree 2 的根节点 “2” 的右子树。
不过需要注意的是,如果两棵子树的根节点为空,则返回空;
如果某棵子树的某个节点为空,则返回另外一棵子树的对应节点;
否则的话,需要将根节点的值相加,将相加的结果放到一棵子树上(或者新建一棵树),递归地对左子树和右子树进行同样的处理。
4. 代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
// 如果两棵子树的根节点均为空,则返回空
if(t1 == null && t2 == null)
return null;
// 若某棵子树的根节点为空,则返回不为空的根节点
else if(t1 == null)
return t2;
else if(t2 == null)
return t1;
// 否则,两棵子树的根节点均不为空,则将值相加,放到 Tree 1 树中
//递归地对左子树和右子树进行同样的处理
else
{
t1.val += t2.val;
t1.left = mergeTrees(t1.left, t2.left);
t1.right = mergeTrees(t1.right, t2.right);
}
return t1;
}
}
5. 验证
6. 出处
- LeeCode 617. 合并二叉树
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-binary-trees
著作权归领扣网络所有。