题目
思路
使用递归建树,流程如下:
-
取出后序节点创建新树的节点
-
找到新树的节点在中序中的索引
-
分割中序序列
-
分割后序序列
-
继续递归建立整颗新树
# 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]) -> Optional[TreeNode]:
# 如果所有节点都已经扫过,说明新的树已经建成了,应返回空节点
if not inorder:
return None
# 1.取出前序节点创建新树的节点
root = TreeNode(preorder[0])
# 2.找到新树的节点在中序中的索引
in_index = inorder.index(root.val)
# 3.分割中序序列
in_left = inorder[:in_index]
in_right = inorder[in_index+1:]
# 4.分割前序序列
pre_left = preorder[1:len(in_left)+1]
pre_right = preorder[len(in_left)+1:]
# 5.继续递归建立整颗新树
root.left = self.buildTree(pre_left, in_left)
root.right = self.buildTree(pre_right, in_right)
return root