利用python 完成leetcode105 从前序与中序遍历序列构造二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:

3

/
9 20
/
15 7
思路
因为前序遍历首先遍历根节点,所以preorder 的第一个元素就是根节点,因为中序遍历先遍历左半部分,再遍历根节点,
所以inorder 中preorder [0]的位置 index 左侧的元素inorder [:index]构成二叉树的左子树, index位置 右侧的元素inorder [index+1:]构成二叉树的右子树
在preorder中,preorder[1:index+1]构成二叉树的左子树,preorder[index+1:]构成二叉树的右子树,preorder[1]为左子树的根节点,preorderr[index+1:]为右子树的根节点,
先利preorder[1:index+1],inorder [:index]构造左子树,再用preorderr[index+1:]和inorder [index+1:]构造右子树,用递归即可
代码

def buildTree(self, preorder, inorder):
        if(len(preorder)==0):return None
        if(len(preorder)==1):return TreeNode(preorder[0])
        N=TreeNode(preorder[0])
        l=inorder.index(preorder[0])
        N.left=self.buildTree(preorder[1:l+1],inorder[:l])
        N.right=self.buildTree(preorder[l+1:],inorder[l+1:])
        return N

你可能感兴趣的:(leetcode,leetcode,中等)