先序遍历构造二叉树(Construct Binary Search Tree from Preorder Traversal)java_leetcode 1008

先序遍历构造二叉树(Construct Binary Search Tree from Preorder Traversal)java_leetcode 1008

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhangjingao/article/details/90379789

题干


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

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

 

示例:

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

 

提示:

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

分析

  这个题主要考察二叉搜索树的构造,二叉搜索树就是,当前节点的值大于左节点的所有节点的值,但是小于右节点的所有节点的值。本题目中给出了先序遍历,其实就是在给出一个根节点后,将各个节点插入到二叉搜索树中,从根节点一次往下遍历。

代码


     public TreeNode bstFromPreorder(int[] preorder) {
        if (preorder == null) {
            return null;
        }
        TreeNode root = new TreeNode(preorder[0]);

        for (int i = 1; i < preorder.length; i++) {
            work(root, preorder[i]);
        }
        return root;
    }

    private void work(TreeNode root, int value) {
        if (value > root.val) {
            if (root.right == null) {
                root.right = new TreeNode(value);
            } else {
                work(root.right, value);
            }
        } else {
            if (root.left == null) {
                root.left = new TreeNode(value);
            } else {
                work(root.left, value);
            }
        }
    }

提交结果

在这里插入图片描述

以下是Java代码,使用递归实现先序遍历构造二叉树: ```java class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int x) { val = x; } } class Solution { public TreeNode buildTree(int[] preorder, int[] inorder) { int preLen = preorder.length; int inLen = inorder.length; // 构建二叉树 return buildTreeHelper(preorder, 0, preLen - 1, inorder, 0, inLen - 1); } private TreeNode buildTreeHelper(int[] preorder, int preStart, int preEnd, int[] inorder, int inStart, int inEnd) { if (preStart > preEnd || inStart > inEnd) { // 递归终止条件 return null; } // 先序遍历的第一个节点为根节点 int rootVal = preorder[preStart]; TreeNode root = new TreeNode(rootVal); // 在中序遍历中找到根节点的位置 int rootIndex = 0; for (int i = inStart; i <= inEnd; i++) { if (inorder[i] == rootVal) { rootIndex = i; break; } } // 递归构建左右子树 int leftSubtreeSize = rootIndex - inStart; root.left = buildTreeHelper(preorder, preStart + 1, preStart + leftSubtreeSize, inorder, inStart, rootIndex - 1); root.right = buildTreeHelper(preorder, preStart + leftSubtreeSize + 1, preEnd, inorder, rootIndex + 1, inEnd); return root; } } ``` 其中,buildTree()方法接受先序遍历和中序遍历的数组作为参数,并返回构建好的二叉树的根节点。buildTreeHelper()方法是递归构建二叉树的辅助函数,它接受先序遍历和中序遍历的数组,以及当前子树的范围作为参数,并返回构建好的子树的根节点。在buildTreeHelper()方法中,首先判断递归终止条件,即先序遍历数组或中序遍历数组为空。然后,找到先序遍历数组的第一个节点作为根节点,并在中序遍历数组中找到根节点的位置。接着,递归构建左右子树,并将它们分别作为当前节点的左右子节点。最后,返回当前节点。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值