leetcode 二叉树

一、重建二叉树

题目描述:输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。假设输入的前序遍历和中序遍历的结果中都不含重复的数字

思路:第一次:前序遍历中的第一个节点为根节点,需要在中序遍历中找到与根节点值相同的节点所在的索引。每次找都遍历一次中序遍历效率低。考虑将中序遍历存在字典中,将中序遍历中的值当做键,索引当做值。

因此在执行递归之前先遍历中序遍历并存在字典中

后续:在递归函数中主要做三件事:1找到当前根节点在中序遍历中的索引   2建立根节点  3建立左右节点的递归,左子树的根节点就是前序遍历中当前root的下一个,右子树的根节点是中序遍历中i-left(左子树的长度)+root+1==前序遍历中的根节点(在递归中一定要考虑结束条件以及返回值)

#定义节点
class TreeNode:
    def __init__(self,x)
        self.value=x
        self.left=None
        self.right=None
class Solution:
    def buildTree(self,preorder,inorder):
        def treeBuild(root,left,right):
            if left>right:return
            node=TreeNode(preorder[root])
            i=dic.get(preorder[root])
            node.left=treeBuild(root+1,left,i-1)
            node.right=treeBuild(i-left+root+1,i+1,right)
            return node

        dic={}
        for i in range(len(inorder)):
            dic[inorder[i]]=i
        return treeBuild(0,0,len(inorder)-1)
    


版权声明:本文为cuicui_ruirui原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。