105. 从前序与中序遍历序列构造二叉树
原始题目链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
解题思路:
构造树的过程使用递归方法,需要考虑递归的结束条件和自身调用。构造的过程和分析先序和中序的过程是一样的,先序遍历的第一个值一定是根节点,然后根据这个节点去中序遍历中查找这个根节点,这样就把一棵树分开成了左子树
根 右子树,然后进行构造,再递归的构造左子树和右子树,具体实现看代码及注释。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
# 定义一个函数,递归的调用
# 参数是树的前序遍历的左右边界和中序遍历的左右边界
def helper(pre_s, pre_e, in_s, in_e):
# 没有节点了
if pre_s > pre_e:
return None
# 前序遍历的第一个节点是根节点
root_node = preorder[pre_s]
# 在中序遍历中查找根节点
root_index = find_in_root_dict[root_node]
# 建立根节点
root = TreeNode(root_node)
# 计算左子树的节点个数:中序遍历中根节点的位置索引 - 中序遍历的最左边的位置索引
left_tree_size = root_index - in_s
# 递归构造左子树:传的参数是左子树的先序遍历的左右边界,左子树的中序遍历的左右边界
# 左子树的先序遍历的左右边界分别是:先序遍历中根节点索引 + 1, 根的索引+左子树的个数
# 左子树的中序遍历的左右边界分别是:中序遍历的左边界, 中序遍历中根节点的位置-1
root.left = helper(pre_s + 1, pre_s + left_tree_size, in_s, root_index - 1)
# 递归构造右子树:传的参数是右子树的先序遍历的左右边界,右子树的中序遍历的左右边界
# 右子树的先序遍历的左右边界分别是:根的索引 + 左子树的个数 + 1,先序遍历的最右边界
# 右子树的中序遍历的左右边界分别是:中序遍历中根节点的索引 + 1, 中序遍历的最右边界
root.right = helper(pre_s + left_tree_size + 1, pre_e, root_index + 1, in_e)
return root
# 访问一次所有的中序遍历结果,存到哈希表中,这样在查找根节点的时候可以优化速度
find_in_root_dict = {node: i for i, node in enumerate(inorder)}
# 先序遍历和中序遍历的元素个数都是一样的,算一个就好了
n = len(preorder)
return helper(0, n - 1, 0, n - 1)
参考文献:
https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/solution/cong-qian-xu-yu-zhong-xu-bian-li-xu-lie-gou-zao-9/