【leetcode千题】106. 从中序与后序遍历序列构造二叉树

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

中序遍历 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