《python算法与数据结构2000讲》0617. 合并二叉树

IT狂飙

于 2024-04-23 21:11:38 发布

阅读量97

点赞数

分类专栏： python数据结构与算法2000讲文章标签：算法数据结构 python

本文链接：https://blog.youkuaiyun.com/liu1813335/article/details/138138708

版权

python数据结构与算法2000讲专栏收录该内容

450 篇文章 ¥9.90 ¥99.00

订阅专栏

这篇博客介绍了如何合并两个二叉树，遵循若节点重叠则相加，空节点则保留非空节点的规则。文章提供了解题思路，强调使用前序遍历来递归构建合并后的二叉树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

《python算法与数据结构2000讲》0617. 合并二叉树

标签：树、深度优先搜索、广度优先搜索、二叉树
难度：简单

题目大意

给定两个二叉树，将两个二叉树合并成一个新的二叉树。合并规则如下：

如果两个二叉树对应节点重叠，则将两个节点的值相加并作为新的二叉树节点。
如果两个二叉树对应节点其中一个为空，另一个不为空，则将不为空的节点左心新的二叉树节点。

最终返回新的二叉树的根节点。

解题思路

利用前序遍历二叉树，并按照规则递归建立二叉树。将其对应节点值相加或者取其中不为空的节点做为新节点。

代码

class Solution:
    def mergeTrees(self, root1: TreeNode, root2: TreeNode) -> TreeNode:
        if not root1:
            return root2
        if not root2:
            return root1

        merged = TreeNode(root1.val + root2.val)
        merged.left = self.mergeTrees(root1.left, root2.left)
        merged.right = self.mergeTrees(root1.right, root2.right)
        return merged