1008. 前序遍历构造二叉搜索树

1008. 前序遍历构造二叉搜索树

原始题目链接：https://leetcode.cn/problems/construct-binary-search-tree-from-preorder-traversal/

给定一个整数数组，它表示BST(即 二叉搜索树 )的 先序遍历 ，构造树并返回其根。

保证 对于给定的测试用例，总是有可能找到具有给定需求的二叉搜索树。

二叉搜索树 是一棵二叉树，其中每个节点， Node.left 的任何后代的值 严格小于 Node.val , Node.right 的任何后代的值 严格大于 Node.val。

二叉树的 前序遍历 首先显示节点的值，然后遍历Node.left，最后遍历Node.right。

解题思路：

前序遍历序列的第一个节点一定是根节点，二叉搜索树的性质，小于根节点的一定是左子树，大于的是右子树的节点，所以可以先找边界，然后递归的构造左右子树。

代码实现：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def bstFromPreorder(self, preorder: List[int]) -> Optional[TreeNode]:
        # 定义一个构建左右子树的函数
        # 参数是子树的左右节点在给定preorde数组中的边界位置
        def construct(l, r):
            if l < r:
                root = TreeNode(preorder[l])
                # 寻找分界点：大于根节点的第一个节点
                boundary = next((i for i in range(l + 1, r) if preorder[i] > root.val), r)
                # 分界点左边的是左子树节点，递归构造左子树
                root.left = construct(l + 1, boundary)
                # 分界点右边是右子树节点，递归构造右子树
                root.right = construct(boundary, r)
                return root
                
        return construct(0, len(preorder))

参考文献：
https://leetcode.cn/problems/construct-binary-search-tree-from-preorder-traversal/solution/yuan-di-di-gui-jian-li-shu-100-by-tuotuoli/