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

一、题目

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

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

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

java - 108. 将有序数组转换为二叉搜索树_第1张图片

java - 108. 将有序数组转换为二叉搜索树_第2张图片

二、思路

     * 1 每次取数组的中间位置作为root,数组中root索引左边的作为左子树,右边作为右子树
     * 2 递归建树,特殊情况需要注意:
     *      当前left == right  说明是同一个数组位置,那么他一定属于叶子节点;
     *      当前mid == left    说明mid左边无子树,因此直接将左孩子置null;
     *      当前mid == right   说明mid右边无子树,因此直接将右孩子置null;

三、代码

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        if (nums.length == 1) return new TreeNode(nums[0]);
        int left = 0, right = nums.length - 1;
        TreeNode root = buildTree(left,right,nums);
        return root;
    }
    public TreeNode buildTree(int left, int right,int[] nums) {
        int mid = (right + left) / 2; // 当前根节点的位置
        if (left == right) return new TreeNode(nums[left]);  // 当前是叶子节点
        if (mid == left)return new TreeNode(nums[mid],null,buildTree(mid+1,right,nums)); // 无左孩子
        if (mid == right) return new TreeNode(nums[mid],buildTree(left,mid-1,nums),null); // 无右孩子
        TreeNode root = new TreeNode(nums[mid],buildTree(left,mid-1,nums),buildTree(mid+1,right,nums)); //左右孩子齐全
        return root;
    }
}

你可能感兴趣的:(leetcode,算法,java,二叉树,分治)