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

643 篇文章 5 订阅

题目
返回与给定前序遍历 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 中的值互不相同

【代码】
在这里插入图片描述

# 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】
在这里插入图片描述

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】
在这里插入图片描述

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)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
抱歉,我是一名语言模型AI,无法为您提供实际编程操作,但可以为您提供伪代码: 1. 定义二搜索节点类 class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int val) { this.val = val; this.left = null; this.right = null; } } 2. 实现二搜索的构建方法 public TreeNode buildTree(int[] preorder, int[] inorder, int preStart, int preEnd, int inStart, int inEnd) { if (preStart > preEnd) { return null; } TreeNode root = new TreeNode(preorder[preStart]); int index = inStart; while (index <= inEnd && inorder[index] != root.val) { index++; } int leftSize = index - inStart; root.left = buildTree(preorder, inorder, preStart + 1, preStart + leftSize, inStart, index - 1); root.right = buildTree(preorder, inorder, preStart + leftSize + 1, preEnd, index + 1, inEnd); return root; } 3. 实现二搜索的遍方法 // public void preorder(TreeNode root, List<Integer> list) { if (root == null) { return; } list.add(root.val); preorder(root.left, list); preorder(root.right, list); } // 中 public void inorder(TreeNode root, List<Integer> list) { if (root == null) { return; } inorder(root.left, list); list.add(root.val); inorder(root.right, list); } // 后 public void postorder(TreeNode root, List<Integer> list) { if (root == null) { return; } postorder(root.left, list); postorder(root.right, list); list.add(root.val); } 4. 将遍结果写入csv文件 public void writeCsvFile(List<Integer> list, String filename) { try { FileWriter writer = new FileWriter(filename); for (int val : list) { writer.append(Integer.toString(val)); writer.append(','); } writer.flush(); writer.close(); } catch (IOException e) { e.printStackTrace(); } } 5. 主函数中调用上述方法 public static void main(String[] args) { int[] preorder = {3, 9, 20, 15, 7}; int[] inorder = {9, 3, 15, 20, 7}; int[] postorder = {9, 15, 7, 20, 3}; Solution solution = new Solution(); TreeNode root = solution.buildTree(preorder, inorder, 0, preorder.length - 1, 0, inorder.length - 1); List<Integer> preorderList = new ArrayList<>(); solution.preorder(root, preorderList); solution.writeCsvFile(preorderList, "preorder.csv"); List<Integer> inorderList = new ArrayList<>(); solution.inorder(root, inorderList); solution.writeCsvFile(inorderList, "inorder.csv"); List<Integer> postorderList = new ArrayList<>(); solution.postorder(root, postorderList); solution.writeCsvFile(postorderList, "postorder.csv"); }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值