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

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

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

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

例如,给出

中序遍历 inorder = [9,3,15,20,7]

后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树:

    3

   / \

  9  20

    /  \

   15   7

思路:

  这道题是做完105题后顺带写的,如果看懂上一题LeetCode从前序与中序遍历序列构造二叉树,

这道题思路也完全一模一样了。在上一题我们通过前序遍历的第一个元素得到根节点,在这一题我们无非就是通过后序遍历的最后一个位置获得根节点,然后同样地去中序遍历中定位,分割出左右来。

  其他步骤就类比着写出了,直接看代码吧。

代码:

class Solution(object):

    def buildTree(self, inorder, postorder):

        #左中右  左右中

        if not inorder or not postorder:return

        mid = postorder[-1]#后序遍历的根节点在最后一个位置

        root = TreeNode(mid)

#分别去两个遍历序列里定位到“左”的区间
        root.left=self.buildTree(inorder[:inorder.index(mid)],postorder[:inorder.index(mid)])

#分别去两个遍历序列里定位到“右”的区间      
        root.right=self.buildTree(inorder[inorder.index(mid)+1:],postorder[inorder.index(mid):-1])

        return root

你可能感兴趣的:(leetcode,算法,职场和发展)