【LeetCode 刷题】二叉树-修改与构造

此博客为《代码随想录》二叉树章节的学习笔记,主要内容为二叉树的修改与构造相关的题目解析。

文章目录

226. 翻转二叉树

题目链接

python 复制代码
class Solution:
    def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        if root is None:
            return
        left = self.invertTree(root.left)
        right = self.invertTree(root.right)
        root.right = left
        root.left = right
        return root
  • 上述代码为后序遍历版本,即先递归处理左、右节点,之后处理中间节点
  • 前序遍历也可正确求解,但中序遍历不可正确求解

105.从前序与中序遍历序列构造二叉树

题目链接

python 复制代码
class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> Optional[TreeNode]:
        if not preorder:
            return None
        left_size = inorder.index(preorder[0])
        left = self.buildTree(preorder[1:1+left_size], inorder[:left_size])
        right = self.buildTree(preorder[1+left_size:], inorder[left_size+1:])
        return TreeNode(preorder[0], left, right)
  • list.index(x) 返回查找对象的索引位置,如果没有找到对象则抛出异常。因此此种构造方法只能处理节点值不重复的二叉树
  • 优化点:
    • 用哈希表预处理 inorder 每个元素的下标,可以 O(1) 查到preorder[0] 在 inorder 的位置,从而 O(1) 知道左子树的大小
    • 把递归参数改成子数组下标区间(左闭右开区间)的左右端点,从而避免复制数组

106.从中序与后序遍历序列构造二叉树

题目链接

python 复制代码
class Solution:
    def buildTree(self, inorder: List[int], postorder: List[int]) -> Optional[TreeNode]:
        if not inorder:
            return None
        left_size = inorder.index(postorder[-1])
        left = self.buildTree(inorder[:left_size], postorder[:left_size])
        right = self.buildTree(inorder[left_size+1:], postorder[left_size:-1])
        return TreeNode(postorder[-1], left, right)
  • 与上题类似,注意左右子树区间范围

654.最大二叉树

题目链接

python 复制代码
class Solution:
    def constructMaximumBinaryTree(self, nums: List[int]) -> Optional[TreeNode]:
        if not nums:
            return None
        left_size = nums.index(max(nums))
        left = self.constructMaximumBinaryTree(nums[0:left_size])
        right = self.constructMaximumBinaryTree(nums[left_size+1:])
        return TreeNode(max(nums), left, right)

617.合并二叉树

题目链接

python 复制代码
class Solution:
    def mergeTrees(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root1:
            return root2
        if not root2:
            return root1
        left = self.mergeTrees(root1.left, root2.left)
        right = self.mergeTrees(root1.right, root2.right)
        return TreeNode(root1.val + root2.val, left, right)
相关推荐
忆杰2 分钟前
统计用户本月的连续登录天数
开发语言·python
东皇太星17 分钟前
python 常见数学公式函数使用详解
python
Musennn18 分钟前
leetcode51.N皇后:回溯算法与冲突检测的核心逻辑
java·数据结构·算法·leetcode
TT哇19 分钟前
【数据结构试题】
java·数据结构
zfj32124 分钟前
算法-动态规划-钢条切割问题
算法·动态规划
m0_640743561 小时前
华为OD-2024年E卷-字符串化繁为简[200分] -- python
python·华为od
卜及中1 小时前
【Python】数据处理工具:Pandas详细指南
开发语言·python·学习·pandas
zmuy1 小时前
114. 二叉树展开为链表
数据结构·链表
Kusunoki_D1 小时前
Python 使用 Requests 模块进行爬虫
爬虫·python
巴基海贼王1 小时前
针对数据仓库方向的大数据算法工程师面试经验总结
大数据·数据仓库·算法