题目描述:给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
注意:高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:
输入:nums = [1,3]
输出:[3,1]
解释:[1,null,3] 和 [3,1] 都是高度平衡二叉搜索树。
解决问题–>递归解决:题中说了要转换为一棵高度平衡的二叉搜索树,并且数组又是排过序的,这就好办了,我们可以使用递归的方式,每次取数组中间的值比如mid作为当前节点,mid前面的值作为他左子树的结点值,mid后面的值作为他右子树的节点值.
代码示例:
import java.util.Arrays;
public class ArrayToTreeNode {
public static void main(String[] args) {
}
// 在原来数组的基础上再次进行调用
public static TreeNode sortedArrayToBST1(int[] nums){
// 递归结束的标志
int left = 0, right = nums.length-1;
if(left > right){
return null;
}
// 找到创建当前中间节点
int mid = (left + right) / 2;
TreeNode root = new TreeNode(nums[mid]);
// 构建左子树的数组
int[] leftArray = Arrays.copyOfRange(nums, left, mid); // 包括左边不包括右面
// 构建右子树的数组
int[] rightArray = Arrays.copyOfRange(nums, mid+1, nums.length);
// 分别进行递归即可
root.left = sortedArrayToBST(leftArray);
root.right = sortedArrayToBST(rightArray);
return root;
}
// 递归方式二, 再次调用一个函数,传入nums, left, right
public static TreeNode sortedArrayToBST(int[] nums) {
return dfs(nums, 0, nums.length-1);
}
public static TreeNode dfs(int[] nums, int left, int right){
if(left > right){
return null;
}
// 总是选择中间位置作为根节点
int mid = (left + right) / 2;
TreeNode root = new TreeNode(nums[mid]);
root.left = dfs(nums, left, mid-1);
root.right = dfs(nums, mid+1, right);
return root;
}
}
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;
}
}
链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree/solution/javadi-gui-fang-shi-jie-jue-by-sdwwld/
来源:力扣(LeetCode)