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

相关推荐
网易独家音乐人Mike Zhou3 小时前
【卡尔曼滤波】数据预测Prediction观测器的理论推导及应用 C语言、Python实现(Kalman Filter)
c语言·python·单片机·物联网·算法·嵌入式·iot
搬砖的小码农_Sky6 小时前
C语言:数组
c语言·数据结构
Swift社区7 小时前
LeetCode - #139 单词拆分
算法·leetcode·职场和发展
Kent_J_Truman7 小时前
greater<>() 、less<>()及运算符 < 重载在排序和堆中的使用
算法
飞滕人生TYF7 小时前
m个数 生成n个数的所有组合 详解
java·递归
先鱼鲨生7 小时前
数据结构——栈、队列
数据结构
一念之坤7 小时前
零基础学Python之数据结构 -- 01篇
数据结构·python
IT 青年8 小时前
数据结构 (1)基本概念和术语
数据结构·算法
熬夜学编程的小王8 小时前
【初阶数据结构篇】双向链表的实现(赋源码)
数据结构·c++·链表·双向链表
Dong雨8 小时前
力扣hot100-->栈/单调栈
算法·leetcode·职场和发展