题目
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。链接
思路
数组有序,二叉搜索树,还要保证平衡。
选择中间数字作为二叉搜索树的根节点,这样分给左右子树的数字个数相同或只相差 1 1 1,可以使得树保持平衡。如果数组长度是奇数,则根节点的选择是唯一的,如果数组长度是偶数,则可以选择中间位置左边的数字作为根节点或者选择中间位置右边的数字作为根节点。正确性证明
那就二分,每次选mid
为根节点,左右子节点递归在[left, mid - 1], [mid + 1, right]
这两个区间内建立,尽可能保证每个节点的左右子节点都存在,直到叶子节点,即区间长度为1结束。
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return helper(nums, 0, nums.length - 1);
}
TreeNode helper(int[]arr, int left, int right){
//当left==right,区间长度为1,即当前节点为叶子节点
//因为它左右的元素都已经建立
if(left <= right){
int mid = left + (right - left) / 2;
TreeNode root = new TreeNode(arr[mid]);
root.left = helper(arr, left, mid - 1);
root.right = helper(arr, mid + 1, right);
return root;
}
return null;
}
}