合并二叉树

题目

给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点。

示例 1:

注意: 合并必须从两个树的根节点开始。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/merge-two-binary-trees
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

• T1、T2合并的三种情况：

1. 同一位置结点都存在→相加
2. 其中一棵树该位置结点为NULL→保留另一棵树结点存在的值
3. 特别地，其中一棵树为空树，合并的结果直接输出另一棵树

学习借鉴

深度优先搜索

同时对两棵树从根结点开始前序遍历，递归分析各左右子树结点。

广度优先搜索

将两棵二叉树的原始结点分别放进两个队列，同时从两个队列中取出结点，判断其左右子结点是否为空：

1. 两棵二叉树的左子结点都不为空，创建合并后的二叉树的左子结点，并将每个二叉树中的左子结点都加入相应的队列；

2. 有一个原始二叉树的左子树为空，此时不需要对非空左子树继续遍历，不需要将左子结点加入队列。

对于右子结点和右子树，处理方法同上。

提交代码

深度优先搜索

# 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

学习总结

1. 列表作堆栈使用的编程方法；
2. 学习深度优先搜索、广度优先搜索的实现原理和方法，对两种搜索的理解还不深入；
3. 搜索解法中递归思想的新的体现。