【leetcode.617】合并二叉树
原创
©著作权归作者所有:来自51CTO博客作者SunAlwaysOnline的原创作品,请联系作者获取转载授权,否则将追究法律责任
一、题目描述
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例 1:
注意: 合并必须从两个树的根节点开始。
二、思路
使用深度优先策略,遍历两颗二叉树的同一个位置,会出现以下情况
两个节点都为null,则合并后的节点也为null
两个节点都不为null,则创建一个新的节点,其值为两个节点之和
其中只有一个节点为null,则合并后的节点为这两个节点中的非空节点
合并完一个节点后,再合并它的左右节点,是一个递归的过程。
代码如下:
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
//对应位置上的节点为空
if (t1 == null) {
return t2;
}
if (t2 == null) {
return t1;
}
//对应位置上的节点都不为空
TreeNode node = new TreeNode(t1.val + t2.val);
//继续合并左右节点
node.left = mergeTrees(t1.left, t2.left);
node.right = mergeTrees(t1.right, t2.right);
return node;
}
提交答案: