Description
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例 1:
输入:
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/ \
4 5
/ \ \
5 4 7
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-binary-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
Solution
前序遍历,先访问根节点把根节点合并了,注意两个return
- 节点空时候return的情况
- 正常情况下的return:当前合并过的新节点要作为父节点的子树。
class Solution:
def mergeTrees(self, t1: TreeNode, t2: TreeNode) -> TreeNode:
if not t1: return t2 # 节点为空的return
if not t2: return t1
def merge(t1, t2):
if not t2: return t1
if not t1: return t2
t1.val += t2.val
t1.left = merge(t1.left, t2.left)
t1.right = merge(t1.right, t2.right)
return t1 # 正常的return
merge(t1, t2)
return t1