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

 

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

注意:
你可以假设树中没有重复的元素。
例如，给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树：
    3
   / \
  9  20
    /  \
   15   7

作者：力扣 (LeetCode)
链接：https://leetcode-cn.com/leetbook/read/data-structure-binary-tree/xo98qt/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
"""

#!/usr/bin/python3
# -*- coding: utf-8 -*-


class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right



# 从中序与后序遍历序列构造二叉树
# 对于后序遍历，最后一个元素必为根
# 我们只要找到最后一个元素在中序遍历中的位置就可以将数组一分为二，
# 左侧为左子树的遍历结果，右侧为右子树
# 对拆分后的两个数组递归的进行上述操作即可
def build_tree_by_in_post(in_order, post_order):
    if not post_order:
        return None
    root = TreeNode(post_order[-1])
    root_index = in_order.index(post_order[-1])
    left_inorder = in_order[:root_index]
    right_inorder = in_order[root_index+1:]
    l_left = len(left_inorder)
    left_postorder = post_order[:l_left]
    right_postorder = post_order[l_left:-1]
    root.left = build_tree_by_in_post(left_inorder, left_postorder)
    root.right = build_tree_by_in_post(right_inorder, right_postorder)
    return root