给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
示例 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,3] 和 [3,1] 都是高度平衡二叉搜索树。
有序数组转为平衡二叉树
问题分析:
若不规定平衡,则可以很多种选择,可以任一节点作为根节点。
有了平衡的限制,数组到二叉树的映射选择就少了很多,要以中间节点作为根节点,减少左右子树的深度差。
根节点左和右子树也是一颗平衡二叉树,因此同样的去在左右子树选择根节点,整个问题的关键就成了根节点的选择。
数组元素为奇数个数时,根节点的选择都是同样的,数组元素为偶数个数时,根节点的下标索引有不同选择:
//向下取整
mid = (start+end)/2;
//向上取整
mid = 1+(start+end)/2;
Java代码实现
/**
* Definition for a binary tree node.
*/
public 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;
}
public static void main(String args[]){
int [ ]nums = {-10,-3,0,5,9};
Solution solution = new Solution();
TreeNode root = solution.sortedArrayToBST(nums);
solution.midOrder(root);
}
}
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
TreeNode root = buildBstHelper(nums,0,nums.length-1);
return root;
}
public TreeNode buildBstHelper(int[] nums,int start,int end){
//递归终止条件
if(start>end){
return null;
}
//可能会溢出
int mid = (start+end)/2;
TreeNode root = new TreeNode(nums[mid]);
root.left = buildBstHelper(nums,start,mid-1);
root.right = buildBstHelper(nums,mid+1,end);
return root;
}
public void midOrder(TreeNode root){
if(root==null){
return;
}
midOrder(root.left);
System.out.println(root.val);
midOrder(root.right);
}
}