Leetcode-剑指07 重建二叉树

最新推荐文章于 2024-10-16 10:13:45 发布

原来是大海

最新推荐文章于 2024-10-16 10:13:45 发布

阅读量134

点赞数

文章标签： leetcode 算法深度优先

本文链接：https://blog.csdn.net/chw1234567/article/details/122953962

版权

在这里插入图片描述

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        def recur(root, left, right):
            if left > right: return                               # 递归终止
            node = TreeNode(preorder[root])                       # 建立根节点
            i = dic[preorder[root]]                               # 划分根节点、左子树、右子树
            node.left = recur(root + 1, left, i - 1)              # 开启左子树递归
            node.right = recur(i - left + root + 1, i + 1, right) # 开启右子树递归
            return node                                           # 回溯返回根节点

        dic, preorder = {}, preorder
        for i in range(len(inorder)):
            dic[inorder[i]] = i
        return recur(0, 0, len(inorder) - 1)