Leetcode_剑指Offer_面试题07. 重建二叉树

不热爱技术只想发财

于 2020-03-29 17:43:28 发布

阅读量221

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_40848065/article/details/105182458

版权

题目：输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

在这里插入图片描述

思路：这道题要用到递归，每次先找到根节点，如何再用递归方法分别构造左右子树。

# 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 preorder==[]:return None
        if len(preorder)==1:return TreeNode(preorder[0])
        else:
            flag=TreeNode(preorder[0])
            flag.left=self.buildTree(preorder[1:inorder.index(preorder[0])+1],inorder[:inorder.index(preorder[0])])
            flag.right=self.buildTree(preorder[inorder.index(preorder[0])+1:],inorder[inorder.index(preorder[0])+1:])
        return flag