剑指offer 面试题7 (重建二叉树) python

题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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根节点

    def reConstructBinaryTree(self, pre, tin):

        # write code here

        if  pre and tin:

            root = TreeNode(pre.pop(0))

            mid = tin.index(root.val)

            root.left = self.reConstructBinaryTree(pre,tin[0:mid])

            root.right = self.reConstructBinaryTree(pre,tin[mid+1:])

            return root

        else:

            return None

 

解题思路:

1.前序的第一个是一个根节点,那么我们可以通过前序的第一个和中序判断出根节点左右子树分别有什么数字。例如:前序1 2 4 7 3 5 6 8 中序 4 7 2 1 5 3 8 6.我们可以知道根节点是 1   左子树有 4 7 2  右子树有 5 3 8 6. 这样我们就可以不断重复并且重新构造出二叉树。

猜你喜欢

转载自blog.csdn.net/chocolate_chuqi/article/details/81071878