利用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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值