题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:
Input: preorder = [-1], inorder = [-1]
Output: [-1]
思路: 分治的思想
主要参考的题解链接
# Definition for a binary tree node.
from typing import List
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:
'''
根据先序遍历是 根|左子树|右子树
中序遍历是 左子树|根|右子树
所以可以首先确定根,然后根的左儿子是左子树的根,右儿子是右子树的根
而左右子树部分可以和最开始判断的方式一样,所以分成了构建左右子树两部分
'''
def gou(root, left, right):
# 这个三个参数的含义:root是在先序遍历中确定的根的位置,
# left和right是在中序遍历中左右边界
if left > right:
return
node = TreeNode(preorder[root])
i = dict_inorder[preorder[root]]
node.left = gou(root+1, left, i-1)
node.right = gou(root+i-left+1, i+1, right)
return node
dict_inorder = {}
for i in range(len(inorder)): # 可以这样设置的原因是inorder和preorder中的数是一样的,只是顺序不一样
dict_inorder[inorder[i]] = i
return gou(0, 0, len(inorder)-1)