106. 从中序与后序遍历序列构造二叉树

描述

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

注意:

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

示例

例如，给出

中序遍历 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, x):#         self.val = x#         self.left = None#         self.right = Noneclass Solution:    def buildTree(self, inorder, postorder):        """        :type inorder: List[int]        :type postorder: List[int]        :rtype: TreeNode        """        if not inorder or len(inorder) == 0:            return None        root = TreeNode(postorder[-1])        idx = inorder.index(postorder[-1])        root.left = self.buildTree(inorder[:idx], postorder[:idx])        root.right = self.buildTree(inorder[idx + 1:], postorder[idx:-1])        return root

GitHub 地址：