LeetCode 617. 合并二叉树
题目描述
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例
输入:
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/ \
4 5
/ \ \
5 4 7
解题思路
递归求解,结束条件是 t1结点 和 t2结点均为空的时候;
创建新的结点
root
,结点值为 t1结点 的值加上 t2结点 的值;
- t1结点为空时,结点值取
0
。- t2同理。
- t1 和 t2 会同时为空,同时为空时已经达到递归结束条件返回了。
递归得到
root
的左子树和右子树;
t1 为空时:
root.left = mergeTrees(null, t2.left); root.right = mergeTrees(null, t2.right);
t2 为空时:
root.left = mergeTrees(t1.left, null); root.right = mergeTrees(t1.right, null);
均不为空时:
root.left = mergeTrees(t1.left, t2.left); root.right = mergeTrees(t1.right, t2.right);
返回 root 即可。
AC
/**
* 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;
}
int val1 = t1 != null ? t1.val : 0;
int val2 = t2 != null ? t2.val : 0;
TreeNode root = new TreeNode(val1 + val2);
if (t1 == null) {
root.left = mergeTrees(null, t2.left);
root.right = mergeTrees(null, t2.right);
} else if (t2 == null) {
root.left = mergeTrees(t1.left, null);
root.right = mergeTrees(t1.right, null);
} else {
root.left = mergeTrees(t1.left, t2.left);
root.right = mergeTrees(t1.right, t2.right);
}
return root;
}
}