根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
思路
我们将buildTree设为递归调用的函数,第一行if not inorder: return None则作为基本结束条件
然后将后序遍历的最后一个数字作为根节点,因为后序遍历是左右根,最后一个就是根节点
接着我们在中序遍历中找到这个节点的索引
我们举个例子,二叉树[3,9,20,24,35,15,7]与题目给的例子相似
那么inorder=[24,9,35,3,15,20,7],postorder=[24,35,9,15,7,20,3]
刚刚开始的root=3,i=3
我们的左子树就是inorder[:3],postorder[:3],也就是[9,24,35]
右子树就是inorder[4:],postorder[3:-1],也就是[20,15,7]
这里我们需要注意一下左右子树的中序后序遍历的列表都是一样的
前面满足了递归的基本结束条件,后面的左右子树满足了递归改变状态向基本结束条件演进,以及递归调用
最后全部执行完了,输出root
/3
/9/20
24/35/15/7
# 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, inorder: List[int], postorder: List[int]) -> TreeNode:
if not inorder: return None
root = TreeNode(postorder[-1])
i = inorder.index(root.val)
root.left = self.buildTree(inorder[:i],postorder[:i])
root.right = self.buildTree(inorder[i+1:],postorder[i:-1])
return root