题目意思很简单:给你两个二叉树,将两个二叉树合并,合并的规则是如果某一个点是重合的,将它们的值相加,在某一个节点二叉树a有值而b没有值,则用a的值作为合并后二叉树在该节点的值。
这道题说难也不难,说不难也能让人头疼,思路上来说很简单,但要实现代码的时候还是略有些繁琐,也不能用繁琐来说,应该用复杂吧。
先上代码:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def mergeTrees(self, t1, t2):
"""
:type t1: TreeNode
:type t2: TreeNode
:rtype: TreeNode
"""
if not t1 and not t2:
return None
ans = TreeNode((t1.val if t1 else 0) + (t2.val if t2 else 0))
ans.left = self.mergeTrees(t1 and t1.left, t2 and t2.left)
ans.right = self.mergeTrees(t1 and t1.right, t2 and t2.right)
return ans
这是一种充分利用了python语言特性的解法,在这里我解释一下这个语法特性,其实我自己也是第一次实践,首先是t1.val if t1 else 0
,这是python的一种简化的语句,if为真,取if前面的值,if为假,取else的值,当然这种话我不说你也看的出来O(∩_∩)O哈哈~。
另一个重点说一下这个t1 and t1.left
,在python中的意思是:如果t1不为None,则取t1.left,如果t1为null,则取t1也就是null,可以类比于其他语言中比较常见的三元运算符等同于t1 == null ? t1 : t1.left
,相信这么写的话,很多人马上就明白了。
另外介绍一种写法,本质上没有任何区别,只是更加容易看明白而已:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def mergeTrees(self, t1, t2):
"""
:type t1: TreeNode
:type t2: TreeNode
:rtype: TreeNode
"""
if not t1 and not t2:
return
if not t1:
res = t2
elif not t2:
res = t1
else:
res = TreeNode(t1.val + t2.val)
res.left = self.mergeTrees(t1.left, t2.left)
res.right = self.mergeTrees(t1.right, t2.right)
return res
两种解法的思路是一样的,只是实现的方式不一样,第一种更具有技巧性。
我个人做题推荐大家多去了解不同的解题思路,以开阔自己的事业,提高自己的思维广度,毕竟不同思维方式的碰撞总是会有一些奇妙的东西呈现。
以上。