LeetCode 105 从前序与中序遍历序列构造二叉树 NC12 重建二叉树

麦格芬230

已于 2022-06-05 21:29:41 修改

阅读量355

点赞数

分类专栏：数据结构与算法文章标签：二叉树数据结构算法 leetcode

于 2020-12-28 00:38:13 首次发布

本文链接：https://blog.csdn.net/weixin_42295205/article/details/111829269

版权

数据结构与算法专栏收录该内容

241 篇文章 0 订阅

订阅专栏

BM40 重建二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

3
/ \
9 20
/ \
15 7

解题思路：

递归

力扣

Python代码：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        if not preorder or not inorder:
            return None

        root_value = preorder[0]
        root = TreeNode(root_value)
        inorder_index = inorder.index(root_value)

        root.left = self.buildTree(preorder[1:inorder_index+1], inorder[:inorder_index])
        root.right = self.buildTree(preorder[inorder_index+1:], inorder[inorder_index+1:])
        return root