如何从前序与中序遍历序列构造python二叉树 - 大数据

【题目】

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

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

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

【思路】

首先回顾遍历的顺序：前序遍历是根节点-左子树-右子树，中序遍历是左子树-根节点-右子树。

那么前序遍历数组的第一个元素肯定是根节点，在中序遍历数组中找到这个元素，则其前一部分是左子树的元素，其后一部分是右子树的元素。递归即可求解。

注意：前序遍历+后序遍历，不能确定唯一的二叉树！

【代码】

python版本

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

class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        # 前序遍历，第一个是head
        # 中序遍历，前一部分是左子树，后一部分是右子树
        if len(preorder) == 0:
            return None
        
        node = TreeNode(preorder[0])
        index = inorder.index(preorder[0])
        node.left = self.buildTree(preorder[1: index + 1], inorder[:index])
        node.right = self.buildTree(preorder[index + 1:], inorder[index + 1:])
        return node

【相似题目】

从中序与后序遍历序列构造二叉树

解题思路：后序遍历数组的最后一个元素是根节点的元素，同样在中序遍历数组中找到该元素，递归生成二叉树。

根据前序和后序遍历构造二叉树

解题思路：直接生成只有右孩子的二叉树即可满足条件。