根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7] 后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3 / \ 9 20 / \ 15 7
思路:
跟105. 从前序与中序遍历序列构造二叉树几乎一样,区别只在于root的值从preorder[0]变成了postorder[-1],其他基本照搬就行。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def buildTree(self, inorder, postorder):
"""
:type inorder: List[int]
:type postorder: List[int]
:rtype: TreeNode
"""
if not postorder:
return None
root = TreeNode(postorder[-1])
root_index = inorder.index(postorder[-1])
left_inorder = inorder[:root_index]
right_inorder = inorder[root_index + 1:]
l_left = len(left_inorder)
left_postorder = postorder[:l_left]
right_postorder = postorder[l_left : -1]
root.left = self.buildTree(left_inorder, left_postorder)
root.right = self.buildTree(right_inorder, right_postorder)
return root