从前序与中序遍历序列构造二叉树Python解法

最新推荐文章于 2024-02-20 09:21:03 发布

皮_客

最新推荐文章于 2024-02-20 09:21:03 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： python leetcode

本文链接：https://blog.csdn.net/qq_43593534/article/details/123865236

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal

例：

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

解析：

使用递归调用，每次递归存储当前子树的根节点。先序遍历的第一个即为根节点。将先序遍历的第一个节点在中序遍历中找出，在中序遍历中此节点的左边为左子树，右边为右子树。记录左子树的长度L_left，先序遍历中第一个节点（也就是根节点）向后L_left长度即为左子树，然后剩余的为右子树。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: