根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 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