【树-中等】1008. 前序遍历构造二叉搜索树

2021年4月24日 100次阅读 来源: 菜菜2022

【题目】
返回与给定前序遍历 preorder 相匹配的二叉搜索树（binary search tree）的根结点。

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

题目保证，对于给定的测试用例，总能找到满足要求的二叉搜索树。

示例：

输入：[8,5,1,7,10,12]
输出：[8,5,10,1,7,null,12]

提示：

1 <= preorder.length <= 100
1 <= preorder[i] <= 10^8
preorder 中的值互不相同

【代码】
《【树-中等】1008. 前序遍历构造二叉搜索树》

# 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 create(self,nums):
        if not nums:
            return None
        root=TreeNode(nums[0])
        high=len(nums)
        i=0
        flag=False
        for i in range(1,high):
            if nums[0]<nums[i]:
                flag=True
                break
        if flag==False:
            i+=1
        root.left=self.create(nums[1:i])
        root.right=self.create(nums[i:])
        return root

    def bstFromPreorder(self, preorder: List[int]) -> TreeNode:
        if not preorder:
            return None
        return self.create(preorder)

【方法2】
《【树-中等】1008. 前序遍历构造二叉搜索树》

class Solution:
    def bstFromPreorder(self, preorder: List[int]) -> TreeNode:
        if preorder:
            root = TreeNode(preorder.pop(0))
            l, r = [], []
            for i in preorder:
                if i <= root.val:
                    l += [i]
                else:
                    r += [i]            
            root.left = self.bstFromPreorder(l)
            root.right = self.bstFromPreorder(r)
            return root

【方法3】
《【树-中等】1008. 前序遍历构造二叉搜索树》

class Solution:
    def bstFromPreorder(self, preorder: List[int]) -> TreeNode:
        def func(q):
            if q:
                p,root=[[],[]],TreeNode(q.pop(0))
                [p[x>root.val].append(x) for x in q]
                root.left=func(p[0])
                root.right=func(p[1])
                return root
        return func(preorder)
    原文作者：菜菜2022
    原文地址: https://blog.csdn.net/kz_java/article/details/119760877
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞