题目
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
解题思路
- 对于该题来讲,要先知道二叉搜索树的定义,简单来讲就是比根节点小的放左边,比根节点大的放右边,并且要平衡,左右高度差不超过一,那么一个有序的数组的中间那个节点做根节点才能保证不超过1.
- 那么递归的思想就出来了,吧数组中间的那个节点为根节点返回,他的左边元素是他的左子树,右边是右子树。所以递归去调用函数。
- ps:如果下面代码看完后,建议可以去搜Arrays工具类的copyOfRange()方法看看。
代码
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
if(nums ==null || nums.length==0){
return null;
}
if(nums.length==1){
return new TreeNode(nums[0]);
}
int mid = nums.length/2;
TreeNode tree = new TreeNode(nums[mid]);
int[] nums1 = new int[mid];
int[] nums2 = new int[nums.length-1-mid];
for(int i=0; i<mid;i++){
nums1[i]=nums[i];
}
for(int i=mid+1; i<nums.length;i++){
nums2[i-mid-1]=nums[i];
}
tree.left= mid==0?null:sortedArrayToBST(nums1);
tree.right= mid==nums.length?null:sortedArrayToBST(nums2);
return tree;
}
}