leetcode617合并二叉树(python实现)

1.问题描述


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

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 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

2.python求解


递归的思想:我们把T1,和T2合并到T1,所以分为以下四种情况:

  • 如果T1、T2都不为空,则把T2的值加到T1上,递归遍历T1、T2的左右孩子都赋给T1的左右孩子
  • 如果T1不为空,T2为空,T1的值不变,递归遍历T1的左右孩子都赋给T1的左右孩子
  • 如果T1为空,T2不为空,以T2结点的值new一个结点做T1,递归遍历T2的左右孩子都赋给T1的左右孩子
  • 如果T1、T2都为空,返回空

代码如下

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def mergeTrees(self, t1, t2):
        """
        :type t1: TreeNode
        :type t2: TreeNode
        :rtype: TreeNode
        """
        if t1 is not None and t2 is not None:
            t1.val += t2.val
            t1.left = self.mergeTrees(t1.left,t2.left)
            t1.right = self.mergeTrees(t1.right,t2.right)
            return t1
        elif t1 is not None and t2 is None:
            t1.val = t1.val
            t1.left = self.mergeTrees(t1.left,None)
            t1.right = self.mergeTrees(t1.right,None)
            return t1
        elif t1 is None and t2 is not None:
            t1 = TreeNode(t2.val)
            t1.left = self.mergeTrees(None,t2.left)
            t1.right = self.mergeTrees(None,t2.right)
            return t1
        else:
            return 

3求解过程遇到的问题


t1.left = self.mergeTrees(None,t2.left)
t1.right = self.mergeTrees(None,t2.right)

在我第一次try的时候,没有把每次递归的结果赋给t1的左右孩子,这样的导致的结果是当t1为空,t2不为空的时候,t2的左右孩子不能合并给t1。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值