105. 从前序与中序遍历序列构造二叉树

105. 从前序与中序遍历序列构造二叉树

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

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

例如，给出

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

通过次数65,024提交次数99,429 在真实的面试中遇到过这道题？

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        if not preorder:
            return None
        root=TreeNode(preorder[0])
        loc=inorder.index(preorder[0])
        root.left=self.buildTree(preorder[1:loc+1],inorder[:loc])
        root.right=self.buildTree(preorder[loc+1:],inorder[loc+1:])
        return root

1. 首先前序遍历的顺序是根->左->右，中序遍历的顺序是左->根->右
2. 这道题的关键其实是到开始遍历右节点之前，前序遍历跟中序遍历所遍历的节点数是相同的。
3. 所以我们在中序遍历中找到的根节点的位置就是分界线，也就是在前序遍历中左右子树的分界线
4. 我们根据上述进行递归就可以求解了
5. 一个原则前序遍历跟中序遍历里面的节点数一定是相同的。

改进

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        dic=dict()
        n=len(preorder)
        for i in range(len(inorder)):
            dic[inorder[i]]=i
        def dfs(pre_start,pre_end,ino_start,ino_end):
            if pre_start==pre_end:
                return None
            root=TreeNode(preorder[pre_start])
            loc=dic[preorder[pre_start]]
            root.left=dfs(pre_start+1,pre_start+loc-ino_start+1,ino_start,loc)
            root.right=dfs(pre_start+loc-ino_start+1,pre_end,loc+1,ino_end)
            return root
        return dfs(0,n,0,n)

1. 可以注意到前面我们差找下标的时候使用的是，find这样消耗的时间比较多，我们改用字典来储存，节省查找的时间。
2. 同时使用记录前后start 和 end的方式代替了切片来节省时间。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        dic=dict()
        n=len(preorder)
        for i in range(len(inorder)):
            dic[inorder[i]]=i
        def dfs(pre_start,pre_end,ino_start,ino_end):
            if pre_start>pre_end:
                return None
            root=TreeNode(preorder[pre_start])
            loc=dic[preorder[pre_start]]
            root.left=dfs(pre_start+1,pre_start+loc-ino_start,ino_start,loc-1)
            root.right=dfs(pre_start+loc-ino_start+1,pre_end,loc+1,ino_end)
            return root
        return dfs(0,n-1,0,n-1)

1. 这个跟上面的代码是一样的就是改变了一下end到底包不包含的情况。