题目:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
代码:
/**
* @作者:dhc
* @创建时间:18:58 2018/7/31
* @描述:108.将有序数组转化为二叉搜索树
*/
public class OneHundredAndEight {
//思路:首先找出序列中位数,作为root结点,然后递归建立,数组的前半部分返回一个二叉搜索树
//这个返回的二叉搜索树的根节点为当前root的左子树,同理右子树
public static TreeNode sortedArrayToBST(int[] nums) {
return getSubNode(nums,0,nums.length - 1);
}
public static TreeNode getSubNode(int[] nums, int l, int r){
//对于只有左结点的结点,它的这个左结点的l==r,因此直接返回(因为是平衡树)
if (r == l) {
return new TreeNode(nums[l]);
}
//一开始没考虑,后来调试,发现对于对于只有左结点的结点,它的l会比r大1,并且最由的那个结点的l==nums.length
//神烦这种条件
if(l > r && r + 1 == l || l == nums.length){
return null;
}
//一开始用(r-l)/2,后来在根结点的右子树上会出错,在右子树这边还需要加上l,递归不好写
int m = (r + l)/2;
TreeNode root = new TreeNode(nums[m]);
root.left = getSubNode(nums,l,m-1);
root.right = getSubNode(nums, m+1, r);
return root;
}
//中序遍历验证结果
public static void midPrint(TreeNode root) {
if(root != null){
midPrint(root.left);
System.out.print(root.val+" ");
midPrint(root.right);
}
}
public static void main(String[] args) {
int[] nums = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
TreeNode root = sortedArrayToBST(nums);
midPrint(root);
}
}