leetcode 1008 Construct Binary Search Tree from Preorder Traversal

leetcode 1008 Construct Binary Search Tree from Preorder Traversal

1.题目描述

返回与给定先序遍历 preorder 相匹配的二叉搜索树(binary search tree)的根结点。

(回想一下,二叉搜索树是二叉树的一种,其每个节点都满足以下规则,对于 node.left 的任何后代,值总 < node.val,而 node.right 的任何后代,值总 > node.val。此外,先序遍历首先显示节点的值,然后遍历 node.left,接着遍历 node.right。)

leetcode 1008 Construct Binary Search Tree from Preorder Traversal
示例:
输入:[8,5,1,7,10,12]
输出:[8,5,10,1,7,null,12]

提示:
1 <= preorder.length <= 100
先序 preorder 中的值是不同的。

2.解题思路

根据二叉搜索树的规则,第一个数字一定是父节点,遍历之后的数字,当出现第一个大于父节点的数字后,这个数字以后的全部在右子树,之前的全部在左子树。递归实现即可。

3.Python代码

class Solution(object):
    def bstFromPreorder(self, preorder):
        """
        :type preorder: List[int]
        :rtype: TreeNode
        """
        if not preorder:
            return None
        root=TreeNode(preorder[0])
        seg=len(preorder)
        for i in range(1,len(preorder)):
            if preorder[i]>root.val:
                seg=i
                break
        root.left=self.bstFromPreorder(preorder[1:seg])
        root.right=self.bstFromPreorder(preorder[seg:])
        return root