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)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值