【LeetCode】108. 将有序数组转换为二叉搜索树（简单）——代码随想录算法训练营Day23

题目链接：108. 将有序数组转换为二叉搜索树

题目描述

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

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例 1：

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

示例 2：

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

提示：

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

文章讲解：代码随想录

视频讲解：构造平衡二叉搜索树！| LeetCode：108.将有序数组转换为二叉搜索树_哔哩哔哩_bilibili

题解1：递归法

思路：以数组中心元素为根节点，以左右两部分元素分别递归的构造左子树和右子树。

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {number[]} nums
 * @return {TreeNode}
 */
var sortedArrayToBST = function(nums) {
    // 左闭右开
    const build = function (left, right) {
        if (left === right) {
            return null;
        }
        const index = left + ((right - left - 1) >> 1); // 取区间中心元素为根节点
        return new TreeNode(nums[index], build(left, index), build(index + 1, right)); // 递归的构造左右子树，并返回根节点
    }
    return build(0, nums.length);
};

分析：时间复杂度为 O(n)，空间复杂度为 O(logn)。

题解2：迭代法

思路：用3个栈（或队列）模拟分割数组的过程，1个队列存储节点，另外2个队列分别存储左下标和右下标。

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {number[]} nums
 * @return {TreeNode}
 */
var sortedArrayToBST = function(nums) {
    const nodeQue = [new TreeNode(0)]; // 创建节点栈
    const leftQue = [0]; // 初始化左区间栈，初始区间左边界为0
    const rightQue = [nums.length]; // 初始化右区间栈，左闭右开区间，初始区间右边界为 nums.length
    const root = new TreeNode(0); // 创建根节点
    nodeQue.push(root);
    while (nodeQue.length > 0) {
        const node = nodeQue.pop();
        const left = leftQue.pop();
        const right = rightQue.pop();
        const index = left + ((right - left - 1) >> 1);
        node.val = nums[index]; // 为当前节点赋值
        if (index > left) {
            // 创建左孩子，并入栈
            node.left = new TreeNode(0);
            nodeQue.push(node.left);
            // 创建区间 [left, index)
            leftQue.push(left);
            rightQue.push(index);
        }
        if (index + 1 < right) {
            // 创建右孩子，并入栈
            node.right = new TreeNode(0);
            nodeQue.push(node.right);
            // 创建区间 [index + 1, right)
            leftQue.push(index + 1);
            rightQue.push(right);
        }
    }
    return root;
};

分析：时间复杂度为 O(n)，空间复杂度为 O(logn)。

收获

又是一道数组构造二叉树的题目，区间定义很重要，一定要统一。