题目
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例 1:
注意: 合并必须从两个树的根节点开始。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-binary-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
- T1、T2合并的三种情况:
- 同一位置结点都存在→相加
- 其中一棵树该位置结点为NULL→保留另一棵树结点存在的值
- 特别地,其中一棵树为空树,合并的结果直接输出另一棵树
学习借鉴
深度优先搜索
同时对两棵树从根结点开始前序遍历,递归分析各左右子树结点。
广度优先搜索
将两棵二叉树的原始结点分别放进两个队列,同时从两个队列中取出结点,判断其左右子结点是否为空:
-
两棵二叉树的左子结点都不为空,创建合并后的二叉树的左子结点,并将每个二叉树中的左子结点都加入相应的队列;
-
有一个原始二叉树的左子树为空,此时不需要对非空左子树继续遍历,不需要将左子结点加入队列。
对于右子结点和右子树,处理方法同上。
提交代码
深度优先搜索
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def mergeTrees(self, t1: TreeNode, t2: TreeNode) -> TreeNode:
if not t1:
return t2
if not t2:
return t1
merged = TreeNode(t1.val + t2.val)
merged.left = self.mergeTrees(t1.left,t2.left)
merged.right = self.mergeTrees(t1.right,t2.right)
return merged
-
时间复杂度:O(min(m,n)),其中 m 和 n 分别是两棵二叉树的结点个数。
两棵树的结点同时不为空才会进行合并相加操作,不会超过较小二叉树的结点数。 -
空间复杂度:O(min(m,n)),其中 m 和 n 分别是两棵二叉树的结点个数。
取决于递归调用的层数,不会超过较小二叉树的最大高度,最坏的情况是,二叉树的高度等于结点数。
广度优先搜索
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def mergeTrees(self, t1: TreeNode, t2: TreeNode) -> TreeNode:
//特别情况,优先考虑
if not t1:
return t2
if not t2:
return t1
//分别加入三个队列
merged = TreeNode(t1.val + t2.val)
queue = collections.deque([merged])
queue1 = collections.deque([t1])
queue2 = collections.deque([t2])
while queue1 and queue2:
node = queue.popleft()
node1 = queue1.popleft()
node2 = queue2.popleft()
left1, right1 = node1.left, node1.right
left2, right2 = node2.left, node2.right
if left1 or left2:
if left1 and left2:
left = TreeNode(left1.val + left2.val)
node.left = left
queue.append(left)
queue1.append(left1)
queue2.append(left2)
elif left1:
node.left = left1
elif left2:
node.left = left2
if right1 or right2:
if right1 and right2:
right = TreeNode(right1.val + right2.val)
node.right = right
queue.append(right)
queue1.append(right1)
queue2.append(right2)
elif right1:
node.right = right1
elif right2:
node.right = right2
return merged
学习总结
- 列表作堆栈使用的编程方法;
- 学习深度优先搜索、广度优先搜索的实现原理和方法,对两种搜索的理解还不深入;
- 搜索解法中递归思想的新的体现。