LeetCode--105. Construct Binary Tree from Preorder and Inorder Traversal(从前序遍历和中序遍历的结果重组二叉树)

题目:

给定两个数组,分别是前序遍历的结果和中序遍历的结果,返回该二叉树。样例如下:


解题思路:

已知前序遍历的第一个值一定是跟节点,在上例中,preorder的第一个值3作为根节点,再把inorder以3为间隔分开,左边为左子树,右边为右子树。而preorder中3之后跟的先是左子树的数值,再是右子树的数值,可以根据inorder得到的左子树和右子树的长度,将preorder除根节点以外,分为左子树的前序遍历和右子树的前序遍历结果。

从上一步可以得到左子树的前序遍历和中序遍历,以及右子树的前序遍历和中序遍历,再重复递归上述步骤即可。

代码:

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

class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        def Iteration(preorder,inorder):
            if len(preorder)==0:
                return None
            if len(preorder)==1:
                return TreeNode(preorder[0])
            root = TreeNode(preorder[0])
            for i in range(len(inorder)):
                if inorder[i]==preorder[0]:
                    break
            preorder0 = preorder[1:1+len(inorder[:i])]
            preorder1 = preorder[1+len(inorder[:i]):]
            root.left = Iteration(preorder0,inorder[:i])
            root.right = Iteration(preorder1,inorder[i+1:])
            return root
        return Iteration(preorder,inorder)

猜你喜欢

转载自blog.csdn.net/xiaoxiaoley/article/details/80319150