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

相关推荐
靖难都1 分钟前
数据结构:单链表
数据结构
perseveranceX10 分钟前
插入排序:扑克牌式的排序算法!
c语言·数据结构·插入排序·时间复杂度·排序稳定性
2501_9387739916 分钟前
文档搜索引擎搜索模块迭代:从基础检索到智能语义匹配升级
人工智能·算法·搜索引擎
CS创新实验室24 分钟前
典型算法题解:长度最小的子数组
数据结构·c++·算法·考研408
我有一些感想……27 分钟前
浅谈 BSGS(Baby-Step Giant-Step 大步小步)算法
c++·算法·数论·离散对数·bsgs
麦麦大数据32 分钟前
F042 A星算法课程推荐(A*算法) | 课程知识图谱|课程推荐vue+flask+neo4j B/S架构前后端分离|课程知识图谱构造
vue.js·算法·知识图谱·neo4j·a星算法·路径推荐·课程推荐
贝塔实验室1 小时前
LDPC 码的度分布
线性代数·算法·数学建模·fpga开发·硬件工程·信息与通信·信号处理
快手技术1 小时前
端到端短视频多目标排序机制框架 EMER 详解
算法
Wenhao.1 小时前
LeetCode LRU缓存
算法·leetcode·缓存·golang
京东零售技术1 小时前
告别 “盲买”!京东 AI 试穿 Oxygen Tryon:让服饰购物从“想象”到“所见即所得”
算法