题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
leetcode原题
思路:
- 前序遍历:根左右
中序遍历:左根右 - python中列表对象的index方法,返回列表中查找元素的索引值
- 根节点在中序遍历序列中的索引为 index,证明:左边有index个元素,而左边的元素均为左子树的节点,所以有左子树有index个元素。
前序遍历 根节点后紧跟左子树的节点,preorder[0]是根节点,其后index个元素是左子树节点。而preorder[1:index+1],左闭右开,内含index个元素
代码:
# Definition for a binary tree node.
# 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:
if len(preorder)==1:
return TreeNode(preorder[0])
elif preorder==[]:
return
else:
root=TreeNode(preorder[0])#【前序遍历序列】中的第一个元素是根节点
index=inorder.index(preorder[0])#找到根节点在【中序遍历序列】中的索引
root.left=self.buildTree(preorder[1:index+1],inorder[0:index])
root.right=self.buildTree(preorder[index+1:],inorder[index+1:])
return root