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

相关推荐
郝学胜-神的一滴26 分钟前
中秋特别篇:使用QtOpenGL和着色器绘制星空与满月
开发语言·c++·算法·软件工程·着色器·中秋
qiuiuiu4131 小时前
CPrimer Plus第十六章C预处理器和C库总结2-qsort函数
java·c语言·算法
JuneXcy2 小时前
C++知识点总结用于打算法
c++·算法·图论
wdfk_prog3 小时前
[Linux]学习笔记系列 -- lib/timerqueue.c Timer Queue Management 高精度定时器的有序数据结构
linux·c语言·数据结构·笔记·单片机·学习·安全
zhuzhuxia⌓‿⌓3 小时前
线性表的顺序和链式存储
数据结构·c++·算法
未知陨落3 小时前
LeetCode:95.编辑距离
算法·leetcode
杨小码不BUG4 小时前
小鱼的数字游戏:C++实现与算法分析(洛谷P1427)
c++·算法·数组·信奥赛·csp-j/s
高山有多高4 小时前
栈:“后进先出” 的艺术,撑起程序世界的底层骨架
c语言·开发语言·数据结构·c++·算法
YouEmbedded4 小时前
解码查找算法与哈希表
数据结构·算法·二分查找·散列表·散列查找·线性查找