目录
- 一、题目
- 二、解法
- 完整代码
一、题目
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]
示例 2:
输入: preorder = [-1], inorder = [-1]
输出: [-1]
提示:
1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder 和 inorder 均 无重复 元素
inorder 均出现在 preorder
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]) -> Optional[TreeNode]:if not preorder:return Noneroot = TreeNode(preorder[0])idx = inorder.index(preorder[0])root.left = self.buildTree(preorder[1 : idx + 1], inorder[ : idx])root.right = self.buildTree(preorder[idx + 1 : ], inorder[idx + 1 : ])return root