Skip to content

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

约 834 字大约 3 分钟

归并

2025-02-25

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

给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 平衡 二叉搜索树。

示例 1:

img.png

输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:

示例 2:

img.png

输入:nums = [1,3]
输出:[3,1]
解释:[1,null,3] 和 [3,1] 都是高度平衡二叉搜索树。

提示:

  • 1 <= nums.length <= 104
  • -104 <= nums[i] <= 104
  • nums严格递增 顺序排列

归并

二叉搜索树的中序遍历是升序序列,题目给定的数组是按照升序排序的有序数组,因此可以确保数组是二叉搜索树的中序遍历序列。

总是选择中间位置左边的数字作为根节点,然后用数组左边的元素构建左子树,数组右边的元素构建右子树。构建好返回根节点, 左边界大于右边界时返回空。根节点只能是中间的两个之一,因为题目要求平衡。选定根节点后,我们要用根节点左边的子数组构建左子树, 右边的子数组构建右子树。这是因为题目给的数组是升序数组,所以他一定是二叉树的中序遍历。

前序构建,先处理根节点,在处理左右子树。选择数组的中间节点作为根节点。用剩下的两个数组创建左右子树。结束条件是数组为空。返回值是创建的二叉搜索树的根节点。

选择中间位置左边的数字作为根节点,则根节点的下标为 mid=(left+right)/2,此处的除法为整数除法。

class Solution {
    // 主函数,接收一个升序数组,返回一颗高度平衡的二叉搜索树
    public TreeNode sortedArrayToBST(int[] nums) {
        // 调用辅助函数,传入数组和左右边界
        return helper(nums, 0, nums.length - 1);
    }

    // 辅助函数,用于构建二叉搜索树
    // 接收一个数组和左右边界,将数组中的元素转化为二叉搜索树,返回根节点
    public TreeNode helper(int[] nums, int left, int right) {
        // 如果左边界大于右边界,说明当前区间为空,返回null
        if (left > right) {
            return null;
        }

        // 总是选择中间位置左边的数字作为根节点
        int mid = (left + right) / 2;
        // 创建根节点
        TreeNode root = new TreeNode(nums[mid]);
        // 递归地构建左子树,左子树的元素应为数组中点左边的元素
        root.left = helper(nums, left, mid - 1);
        // 递归地构建右子树,右子树的元素应为数组中点右边的元素
        root.right = helper(nums, mid + 1, right);
        // 返回构建好的根节点
        return root;
    }
}
  • 时间复杂度: O(n)O(n) ,其中 nn 是数组的长度。每个数字只访问一次。
  • 空间复杂度: O(logn)O(\log n) ,其中 nn 是数组的长度。空间复杂度不考虑返回值,因此空间复杂度主要取决于递归栈的深度,递归栈的深度是 O(logn)O(\log n)