Leetcode:105. 从前序与中序遍历序列构造二叉树

105. 从前序与中序遍历序列构造二叉树


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

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

例如,给出

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

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

Python 实现

# 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, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        if len(preorder) == 0 or len(inorder) == 0:
            return None
        mid = inorder.index(preorder[0])
        node =  TreeNode(preorder.pop(0))
        node.left = self.buildTree(preorder[:mid],inorder[:mid])
        node.right = self.buildTree(preorder[mid:],inorder[mid+1:])
        return node
        
posted @ 2018-09-22 00:01  罗梁  阅读(144)  评论(0编辑  收藏  举报