617. Merge Two Binary Trees 深度优先算法+递归

最新推荐文章于 2024-09-12 18:58:16 发布

向光光同学

最新推荐文章于 2024-09-12 18:58:16 发布

阅读量208

点赞数

分类专栏： leetcode 文章标签：深度优先算法 leetcode

本文链接：https://blog.csdn.net/Xg_316167164/article/details/123187322

版权

leetcode 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

在这里插入图片描述

题目搬运者

https://leetcode-cn.com/problems/merge-two-binary-trees/
已知两棵二叉树，root1和root2。
想象一下，当您将其中一个节点覆盖另一个节点时，这两棵树中的一些节点会重叠，而其他节点则不会。您需要将这两棵树合并成一个新的二叉树。合并规则是，如果两个节点重叠，则将节点值相加，作为合并节点的新值。否则，将使用NOT null节点作为新树的节点。
返回合并的树。
注意:合并过程必须从两棵树的根节点开始

思路

深度优先算法+递归

难点忘记了树的对象怎么创建，憨憨一个，果然要多练习敲代码啊，要不然像个精神巨人行动矮子。

在用到深度优选算法和递归的话，代码是很简单啊，思路也很清晰，但是递归套用，时间还是比较慢。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def mergeTrees(self, root1, root2):
        """
        :type root1: TreeNode
        :type root2: TreeNode
        :rtype: TreeNode
        """
        if root1 == None:  return root2
        if root2 == None: return root1
        newTree = TreeNode(root1.val+root2.val)
        newTree.left = Solution.mergeTrees(self,root1.left,root2.left)
        newTree.right = Solution.mergeTrees(self,root1.right,root2.right)
        return newTree

广度优先算法+队列

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def mergeTrees(self, root1, root2):
        """
        :type root1: TreeNode
        :type root2: TreeNode
        :rtype: TreeNode
        """
        if root1 == None:  return root2
        if root2 == None: return root1
       
        newTree = TreeNode(root1.val+root2.val)
        queroot = collections.deque([newTree])
        queroot1 = collections.deque([root1])
        queroot2 = collections.deque([root2])
        
        while queroot1 and queroot2:
            node = queroot.popleft()
            node1 = queroot1.popleft()
            node2 =queroot2.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
                    queroot.append(left)
                    queroot1.append(left1)
                    queroot2.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
                    queroot.append(right)
                    queroot1.append(right1)
                    queroot2.append(right2)
                elif right1:
                    node.right=right1
                elif right2:
                    node.right=right2

        return newTree