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/