LeetCode 108. 将有序数组转换为二叉搜索树 学习笔记（Java）

一、题目

• 原题链接

• 描述：
  给你一个整数数组 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,3] 和 [3,1] 都是高度平衡二叉搜索树。

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

二、题解

• 思路：
  1.递归；
  2.找到中间结点，此节点的左子树为sortedArrayToBST(左边半个数组)，右子树为sortedArrayToBST(右边半个数组)；
  3.边界情况：当nums长度为1和2的时候直接返回二叉树。
• 注意点：
  1.注意左、右一半数组的长度；
  2.答案并不唯一。

三、代码（Java）

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
    	// 只有根节点的树
        if(nums.length==1){
            TreeNode treeNode = new TreeNode(nums[0]);
            return treeNode;
        }
        // 一个根节点和一个叶子节点
        if(nums.length==2){
            TreeNode treeNode = new TreeNode(nums[0]);
            treeNode.right = new TreeNode(nums[1]);
            return treeNode;
        }
        // nums长度为三的情况并不需要单独写出来
        // if(nums.length==3){
        //     TreeNode treeNode = new TreeNode(nums[1]);
        //     treeNode.left = new TreeNode(nums[0]);
        //     treeNode.right = new TreeNode(nums[2]);
        //     return treeNode;
        // }

		// 以中间的数创建根节点
        TreeNode treeNode = new TreeNode(nums[(nums.length-1)/2]);
        // 左边一半数组
        int[] leftArray = new int[(nums.length-1)/2];
        for(int i=0; i<leftArray.length; i++) leftArray[i] = nums[i];
        // 右边一半数组
        int[] rightArray = new int[nums.length-(nums.length-1)/2-1];
        for(int i=0; i<rightArray.length; i++) rightArray[i] = nums[i+(nums.length-1)/2+1];
        // 根节点的左子树就是左边一半数组的二叉树
        treeNode.left = sortedArrayToBST(leftArray);
        // 根节点的右子树就是右边一半数组的二叉树
        treeNode.right = sortedArrayToBST(rightArray);
        return treeNode;
    }
}