思路:递归
二叉搜索树(BST) : 根节点左子树的值都小于根节点的值,根节点右子树的值都大于根节点的值。
这和题中的排序数组有什么关联呢?
如果选定一个值作为根节点,那么数组中,左侧的元素即左子树,右侧的元素即右子树,而左侧的树也是排序的,所以构造左子树的过程和原问题一致,根据对称性,右子树同理。
这样就构造出一个子问题:假设选取数组中的一个元素作为根节点,那么对于该元素左侧的元素递归构成左子树,对于该元素右侧的元素递归构成右子树,现在如何选取根节点成了唯一的问题!
排序数组到BST的映射不是唯一的!题目要求是平衡树,选取数组的中间元素作为根节点!
代码实现:
1.定义递归函数的出口 2.选取中间元素作为根节点,由于奇偶性,始终选择向下取整 3.利用递归解决子问题,即利用数组当前元素的左侧元素们构造左子树,和利用数组当前元素的右侧元素们构造右子树,构造方式和原问题一致
——时间复杂度:O(N) 访问数组中每一个元素一次
——空间复杂度: O(N) 递归的深度是O(logN),并且需要开辟O(N)的空间储存树的所有节点
/**
* 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;
* }
* }
*/
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return helper(nums, 0, nums.length - 1);
}
public TreeNode helper(int[] nums, int left, int right) {
if (left > right) {
return null;
}
// 总是选择中间位置左边的数字作为根节点
int mid = (left + right) / 2;
TreeNode root = new TreeNode(nums[mid]);
root.left = helper(nums, left, mid - 1);
root.right = helper(nums, mid + 1, right);
return root;
}
}