重建二叉树

Posted 2021-02-10 277223178dudu

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回构造的TreeNode根节点
    #采用递归的思想。根据前序遍历和中序遍历来构建二叉树的思路：前序遍历的第一个则是二叉树的根，
    #找到根在中序遍历中的位置，则根将中序遍历分为了两部分，根的左边为二叉树的左子树，
    #根的右边为二叉树的右子树，对应左右子树在可前序遍历中是连续存在的，
    #根据该思路可以继续分别寻找左子树与右子树的根节点，递归进行。
    #判断二叉树是否为空，进行长度的判断即可。
    #还要注意将根定义成节点的形式。在进行本函数的递归调用时，需要在本函数名前面加上self.。
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        if len(pre) == 0:
            return None
        root_data = TreeNode(pre[0])
        i=tin.index(pre[0])
        root_data.left =self.reConstructBinaryTree(pre[1:1+i],tin[:i])
        root_data.right =self.reConstructBinaryTree(pre[1+i:],tin[1+i:])
        return root_data