心路历程:
第一眼看这个题有点懵,后来根据题目给的案例模拟的时候发现,其实只需要在递归函数中每一层构建一个根节点和它的左右子树即可,递归函数传入的是以当前为根节点的前序和中序遍历的数组。
前序遍历用于确认根节点,中序遍历用于确认该根节点的左右子树的长度,然后即可递归不断向下进行。
注意的点:
1、利用递归函数创建二叉树时,每一层new一个node,然后node.left = dfs(...)...,记得最后要把node给return;
2、有return值的递归函数在边界条件和函数上都需要return
3、传参想不明白索引的时候传入数组即可
解法:递归+二叉树建树
python
# 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]:
def dfs(prelist, inlist): # 构建以prelist[0]为根节点的树
if not prelist and not inlist: return None
root = TreeNode(prelist[0])
root_i = inlist.index(prelist[0]) # 因为值是唯一确定的
left_len, right_len = root_i, len(inlist) - root_i
root.left = dfs(prelist[1: 1 +left_len], inlist[:root_i] )
root.right = dfs(prelist[1+left_len:], inlist[root_i+1:] )
return root
return dfs(preorder, inorder)