https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
preorder [ 3 / 9 / 20 15 7 / ]
inorder [ / 9 / 3 / 15 20 7 / ]
preorder[0]就是根节点,此时inorder中该节点的左部分是属于左子树,右部分属于右子树,根据该节点在inorder中的位置pos去划分inorder和preorder。
用递归,终止条件是当preorder为空,返回空,preorder只含一个元素,返回包含该元素的叶子节点TreeNode。根据传入的preorder,inorder重复做一样的事,得到根节点和孩子节点。
class Solution:
def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
if len(preorder) == 0:
return
if len(preorder) == 1:
return TreeNode(preorder[0])
root = TreeNode(preorder[0])
pos = inorder.index(root.val)
if preorder[