LeetCode105题:从前序与中序遍历构造二叉树(python3)

105题:从前序与中序遍历构造二叉树

python 复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> Optional[TreeNode]:
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[1+left_size:])
        return TreeNode(preorder[0],left,right)

时间复杂度:n的平方,其中 n 为 preorder的长度。最坏情况下二叉树是一条链,我们需要递归 O(n)次,每次都需要 O(n)的时间查找 preorder[0]和复制数组。

空间复杂度:n的平方

python 复制代码
class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> Optional[TreeNode]:
        index = {x:i for i,x in enumerate(inorder)}
        def dfs(pre_l:int,pre_r:int,in_l:int,in_r:int) -> Optional[TreeNode]:
            if pre_l == pre_r:
                return None
            left_size = index[preorder[pre_l]] - in_l #左子树大小
            left = dfs(pre_l + 1,pre_l+1+left_size,in_l,in_l+left_size)
            right= dfs(pre_l+1+left_size,pre_r,in_l+1+left_size,in_r)
            return TreeNode(preorder[pre_l],left,right)
        return dfs(0,len(preorder),0,len(inorder)) #左闭右开区间

速度快了很多41ms

相关推荐
小指纹36 分钟前
图论-最短路 Bellman-Ford算法
c++·算法·objective-c·图论
屁股割了还要学36 分钟前
【数据结构入门】时间、空间复杂度的计算
c语言·开发语言·数据结构·c++·算法
myjs9991 小时前
数学=符号
java·前端·算法
麦兜*1 小时前
大模型时代,Transformer 架构中的核心注意力机制算法详解与优化实践
jvm·后端·深度学习·算法·spring·spring cloud·transformer
coding随想1 小时前
算法界的“达摩克利斯之剑”——NP完全性理论
算法
秋难降2 小时前
栈:从基础概念到实战解题(详细)
数据结构·算法·排序算法
归云鹤2 小时前
QT信号和槽怎么传输自己定义的数据结构
开发语言·数据结构·qt
ezl1fe3 小时前
RAG 每日一技(十):向量检索的“死穴”?用混合搜索(Hybrid Search)来拯救!
后端·算法
Ahu_iii3 小时前
【图论基础】理解图的“闭环”:Tarjan 强连通分量算法全解析
算法·图论
PixelMind3 小时前
【IQA技术专题】DISTS代码讲解
图像处理·人工智能·python·算法·iqa