根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
解题思路:中序:左根右,后序:左右根,先由后序遍历找根节点,然后 在中序遍历中找根节点位置,中序遍历左边的为左节点,右边的为右节点,递归构造整个二叉树
# 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, inorder: List[int], postorder: List[int]) -> TreeNode:
length = len(inorder)
if not length:
return None
root = TreeNode(postorder[-1])
i = inorder.index(postorder[-1])
if i:
root.left = self.buildTree(inorder[:i], postorder[:i])
if length - i:
root.right = self.buildTree(inorder[i+1:], postorder[i:length-1])
return root