原题链接
题解
二叉搜索树的中序遍历是升序序列,题目给定的数组是按照升序的有序数组,因此可以确保数组是二叉搜索树的中序遍历序列。
题目中要保证是平衡的二叉搜索树,所以我们只要每次选择数组区间中间的点作为节点,节点的左边的数字一定是该节点的左子树,同理右边也是,就能保证是一个平衡的二叉树,利用递归来实现
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* sortedArrayToBST(vector<int>& nums) {
return build(nums, 0, nums.size() - 1);//这里要注意,我们的区间是左闭右开还是左闭右闭(不同的方式,下面递归的条件和区间的分配是不一样的)
}
TreeNode* build(vector<int>& nums, int l, int r){
if(l > r) return NULL;
int mid = (l + r) >> 1;
TreeNode* tree = new TreeNode(nums[mid]);
tree->left = build(nums, l, mid - 1);
tree->right = build(nums, mid + 1, r);
return tree;
}
};
来个Java版本的
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return build(nums, 0, nums.length - 1);
}
public TreeNode build(int[] nums, int l, int r){
if(l > r) return null;
int mid = (l + r) >> 1;
TreeNode tree = new TreeNode(nums[mid]);
tree.left = build(nums, l, mid - 1);
tree.right = build(nums, mid + 1, r);
return tree;
}
}