LeetCode 617. 合并二叉树

tech2022-08-07  139

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
最新回复(0)