【leetcode】#数组【Python】105. Construct Binary Tree from Preorder and Inorder Traversal 从前序与中序遍历序列构造二叉树

链接:

https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

题目:

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

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

例如,给出

前序遍历 preorder = [3,9,20,15,7] 中序遍历 inorder = [9,3,15,20,7] 返回如下的二叉树:

    3
   /  \
 9     20
      /  \
     15   7

我的解法:标准思路

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def buildTree(self, preorder, inorder):
        if not preorder:
            return None
        node = TreeNode(preorder[0])
        idx = inorder.index(preorder[0])
        node.left = self.buildTree(preorder[1:idx+1], inorder[:idx])
        node.right = self.buildTree(preorder[idx+1:], inorder[idx+1:])
        return node        

别人的解法1:思路是类似的,实现上有小捷径,但是我个人认为不推荐。
每次遍历把左节点一个个pop出去,再算右节点

def buildTree(self, preorder, inorder):
    if inorder:
        print "preorder, inorder",preorder, inorder
        ind = inorder.index(preorder.pop(0))
        root = TreeNode(inorder[ind])
        root.left = self.buildTree(preorder, inorder[0:ind])
        root.right = self.buildTree(preorder, inorder[ind+1:])
        return root

别人的解法2: 时间秒杀前两种的解法,reverse作用就是可以直接pop

def buildTree(self, preorder, inorder):
    def build(stop):
        if inorder and inorder[-1] != stop:
            root = TreeNode(preorder.pop())
            root.left = build(root.val)
            inorder.pop()
            root.right = build(stop)
            return root
    preorder.reverse()
    inorder.reverse()
    return build(None)

你可能感兴趣的:(【leetcode】#数组)