题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例:

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)

这篇博客通过分治思想解析了如何利用先序遍历和中序遍历的结果来构建二叉树。给出了一种递归解决方案,首先确定根节点,然后分别构建左右子树。具体实现中,通过建立中序遍历的字典映射,找到根节点在中序遍历中的位置,以此划分左右子树的范围,进而递归构造整个树结构。
697

被折叠的 条评论
为什么被折叠?



