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

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

题目说明

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

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

提示：

1 <= nums.length <= 10^4
-10^4 <= nums[i] <= 10^4
nums 按 严格递增 顺序排列

示例

示例 1:

1
2
3

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

示例 2:

1
2
3

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

笔者理解

此题是一道二叉树算法问题，在力扣题库中被定义为简单题。

解法

当笔者阅读完此题后，发现我们只需要每次去找区间中间的值作为根节点，再左右区间作为左右子树即可，让我们来看看具体如何实现的吧。

实现

/**
 * Definition for a binary tree node.
 * 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;
 *     }
 * }
 */
class Solution {
    /**
     * 递归
     * 二分
     */
    public TreeNode sortedArrayToBST(int[] nums) {
        int n = nums.length;

        if (n == 0) {
            return null;
        }

        TreeNode result = buildTree(nums, 0, n);

        return result;
    }

    public TreeNode buildTree(int[] nums, int left, int right) {
        if (left == right) {
            return null;
        }

        // 找出区间的中点作为根节点
        int mid = left + (right - left) / 2;
        TreeNode node = new TreeNode(nums[mid]);

        // 将区间二分分别作为左右子树
        node.left = buildTree(nums, left, mid);
        node.right = buildTree(nums, mid + 1, right);

        return node;
    }
}