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

题目描述

将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

给定有序数组: [-10,-3,0,5,9],

一个可能的答案是：[0,-3,9,-10,null,5]，它可以表示下面这个高度平衡二叉搜索树：

      0
     / \
   -3   9
   /   /
 -10  5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

我的思路

因为数组是按照升序排列的，二叉树要求根结点左子树的所有元素都小于根结点元素，右子树的所有元素都大于根结点元素，则只需把数组中间的元素放在根节点，中间左边的所有元素放在左子树，中间右边的所有元素放在右子树，即可保证左右子树的高度差不超过1。

我的程序（Java）

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return nums == null ? null : buildTree(nums, 0, nums.length - 1);
    }
    public TreeNode buildTree(int[] nums, int lo, int hi){
        if(lo > hi)    return null;
        // 把数组中间的数作为根结点
        int mid = lo + (hi - lo)/2;
        TreeNode root = new TreeNode(nums[mid]);
        // 把数组中间的数左边的数放在左子树
        root.left = buildTree(nums, lo, mid - 1);
        // 把数组中间的数右边的数放在右子树        
        root.right = buildTree(nums, mid + 1, hi);
        return root;
    }
}