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

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 = None

class 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 地址： https://github.com/protea-ban/LeetCode