LeetCode高频100题刷题记录之——合并二叉树

1 问题描述

给定两棵二叉树，从根结点开始，将这两棵二叉树合并。合并规则是，重合部分的结点值为两棵树同一位置结点之和，不重合部分由存在的那棵树的结点替换。

2 代码实现

还是比较简单的一道题目，我们只需要定义两个指针，遍历两棵树的所有结点，然后使用递归进行修改即可。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def mergeTrees(self, root1: TreeNode, root2: TreeNode) -> TreeNode:
        if root1 is None:
            return root2
        elif root2 is None:
            return root1
        else:
            p1 = root1
            p2 = root2
            p1.val = p1.val + p2.val
            p1.left = self.mergeTrees(p1.left, p2.left)
            p1.right = self.mergeTrees(p1.right, p2.right)
            return root1

时间复杂度$O(min(n1,n2))$，空间复杂度$O(min(h1,h2))$。

广度优先搜索可以实现，但是需要构造队列和栈存储结点，相对来说比较麻烦，在此省略。