108. 将有序数组转换为二叉搜索树

https://leetcode.cn/problems/convert-sorted-array-to-binary-search-tree/description/?envType=study-plan-v2&envId=top-interview-150

这题考察的就是平衡二叉树的构建,所以需要掌握这部分的知识,如果对这方面知识有所欠缺的话可以看一下这篇博客:二叉搜索树/平衡二叉搜索树-CSDN博客

推销一下自己写的文章,我觉得我写的还是很详细的,认真看完的话这题就没什么问题了。

class Solution {
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode() {}
        TreeNode(int val) { this.val = val; }
        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }
    public TreeNode sortedArrayToBST(int[] nums) {
        TreeNode root = null;
        for(int i = 0; i < nums.length; i++) {
            root = addNode(root, nums[i]);
        }
        return root;
    }

    //计算树的高度
    public int height(TreeNode node) {//叶节点的高度为1
        if(node == null) return 0;
        return Math.max(height(node.left), height(node.right)) + 1;
    }
    //计算平衡因子(左子树的高度 - 右子树的高度)
    public int balanceFactor(TreeNode node) {
        return height(node.left) - height(node.right);
    }

    //左旋
    public TreeNode leftRotate(TreeNode node) {
        TreeNode right = node.right;
        node.right = null; // 断开连接,不然会产生环
        if(right.left != null) node.right = right.left;
        right.left = node;
        return right;
    }
    //右旋
    public TreeNode rightRotate(TreeNode node) {
        TreeNode left = node.left;
        node.left = null; // 断开连接,不然会产生环
        if(left.right != null) node.left = left.right;
        left.right = node;
        return left;
    }

    //添加节点
    public TreeNode addNode(TreeNode node, int val) {
        if(node == null) {
            return new TreeNode(val);
        }
        if(val < node.val) node.left = addNode(node.left, val);
        else if(val > node.val) node.right = addNode(node.right, val);
        // 插入节点后检查是否需要平衡调整
        int balanceFactor = balanceFactor(node);
        if(balanceFactor > 1 && val < node.left.val) {//LL情况->右旋,node是被破坏节点,加入节点是破坏节点
            return rightRotate(node);
        }
        if(balanceFactor < -1 && val > node.right.val) {//RR情况->左旋
            return leftRotate(node);
        }
        if(balanceFactor > 1 && val > node.left.val) {//LR情况->被破坏节点的左节点左旋,被破坏节点右旋
            node.left = leftRotate(node.left);
            return rightRotate(node);
        }
        if(balanceFactor < -1 && val < node.right.val) {//RL情况->被破坏节点的右节点右旋,被破坏节点左旋
            node.right = rightRotate(node.right);
            return leftRotate(node);
        }
        return node;
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        int[] nums = {-10,-3,0,5,9};
        solution.sortedArrayToBST(nums);
    }
}

以下是使用 Kotlin 实现将有序数组转换二叉搜索树的代码: ```kotlin class TreeNode(var `val`: Int) { var left: TreeNode? = null var right: TreeNode? = null } fun sortedArrayToBST(nums: IntArray): TreeNode? { return buildBST(nums, 0, nums.size - 1) } fun buildBST(nums: IntArray, left: Int, right: Int): TreeNode? { if (left > right) { return null } val mid = (left + right) / 2 val root = TreeNode(nums[mid]) root.left = buildBST(nums, left, mid - 1) root.right = buildBST(nums, mid + 1, right) return root } ``` 这段代码中,我们首先定义了一个 `TreeNode` 类,用于表示二叉树的节点。然后,我们定义了一个 `sortedArrayToBST` 函数,该函数接受一个有序整数数组 `nums`,并返回一个高度平衡的二叉搜索树。 在 `sortedArrayToBST` 函数中,我们调用了 `buildBST` 函数,该函数用于递归构建二叉搜索树。`buildBST` 函数接受三个参数:有序整数数组 `nums`、左边界 `left` 和右边界 `right`。我们首先判断左边界是否大于右边界,若是,则返回 `null`。否则,我们选取中间位置 `mid`,将 `nums[mid]` 作为当前节点的值,然后递归构建当前节点的左子树和右子树,并将它们分别赋值给当前节点的 `left` 和 `right` 属性。最后,返回当前节点。 这样,我们就完成了将有序数组转换二叉搜索树的实现。在上述示例中,我们可以调用 `sortedArrayToBST(intArrayOf(-10, -3, 0, 5, 9))`,得到一个高度平衡的二叉搜索树 `[0, -3, 9, -10, null, 5]`。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值