LeetCode-617. Merge Two Binary Trees

来源:互联网 发布:唯品会秒杀软件 编辑:程序博客网 时间:2024/06/08 08:03

题目链接:617. Merge Two Binary Trees


题目意思很简单:给你两个二叉树,将两个二叉树合并,合并的规则是如果某一个点是重合的,将它们的值相加,在某一个节点二叉树a有值而b没有值,则用a的值作为合并后二叉树在该节点的值。

这道题说难也不难,说不难也能让人头疼,思路上来说很简单,但要实现代码的时候还是略有些繁琐,也不能用繁琐来说,应该用复杂吧。

先上代码:

# Definition for a binary tree node.# class TreeNode(object):#     def __init__(self, x):#         self.val = x#         self.left = None#         self.right = Noneclass 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 = Noneclass 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

两种解法的思路是一样的,只是实现的方式不一样,第一种更具有技巧性。
我个人做题推荐大家多去了解不同的解题思路,以开阔自己的事业,提高自己的思维广度,毕竟不同思维方式的碰撞总是会有一些奇妙的东西呈现。

以上。

原创粉丝点击