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

心路历程:

第一眼看这个题有点懵,后来根据题目给的案例模拟的时候发现,其实只需要在递归函数中每一层构建一个根节点和它的左右子树即可,递归函数传入的是以当前为根节点的前序和中序遍历的数组。

前序遍历用于确认根节点,中序遍历用于确认该根节点的左右子树的长度,然后即可递归不断向下进行。

注意的点:

1、利用递归函数创建二叉树时,每一层new一个node,然后node.left = dfs(...)...,记得最后要把node给return;

2、有return值的递归函数在边界条件和函数上都需要return

3、传参想不明白索引的时候传入数组即可

解法:递归+二叉树建树

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]:

        def dfs(prelist, inlist):  # 构建以prelist[0]为根节点的树
            if not prelist and not inlist: return None
            root = TreeNode(prelist[0])
            root_i = inlist.index(prelist[0])  # 因为值是唯一确定的
            left_len, right_len = root_i, len(inlist) - root_i
            root.left = dfs(prelist[1: 1 +left_len], inlist[:root_i] )
            root.right = dfs(prelist[1+left_len:], inlist[root_i+1:] )
            return root

        return  dfs(preorder, inorder)
相关推荐
power 雀儿4 分钟前
Scaled Dot-Product Attention 分数计算 C++
算法
琹箐34 分钟前
最大堆和最小堆 实现思路
java·开发语言·算法
renhongxia11 小时前
如何基于知识图谱进行故障原因、事故原因推理,需要用到哪些算法
人工智能·深度学习·算法·机器学习·自然语言处理·transformer·知识图谱
坚持就完事了1 小时前
数据结构之树(Java实现)
java·算法
算法备案代理1 小时前
大模型备案与算法备案,企业该如何选择?
人工智能·算法·大模型·算法备案
赛姐在努力.2 小时前
【拓扑排序】-- 算法原理讲解,及实现拓扑排序,附赠热门例题
java·算法·图论
野犬寒鸦3 小时前
从零起步学习并发编程 || 第六章:ReentrantLock与synchronized 的辨析及运用
java·服务器·数据库·后端·学习·算法
霖霖总总3 小时前
[小技巧66]当自增主键耗尽:MySQL 主键溢出问题深度解析与雪花算法替代方案
mysql·算法
rainbow68893 小时前
深入解析C++STL:map与set底层奥秘
java·数据结构·算法
wangjialelele3 小时前
平衡二叉搜索树:AVL树和红黑树
java·c语言·开发语言·数据结构·c++·算法·深度优先