题目:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
大体思路:
刚看到这道题的时候开始的思路为:先创建一个头结点,然后依次往里插入,并通过左旋右旋的方式调整。后来感觉这样的话并没有充分利用有序数组的条件,仔细一看这nm不是二分搜索嘛。
选择数组中最中间的元素作为根结点并划分数组,左边的作为左子树,右边的作为右子树。对于左/右子树部分依然按照该方式划分,一个简单的递归即可完成。不过需要注意递归的结束条件应该是输入的数组为空。
实现代码:
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return selution(nums, 0, nums.length - 1);
}
public TreeNode selution(int[] nums, int left, int right){
if (left > right){
return null;
}
int mid = left + (right - left) / 2;
TreeNode current = new TreeNode(nums[mid]);
current.left = selution(nums, left, mid - 1);
current.right = selution(nums, mid + 1, right);
return current;
}
}