将有序数组转换为二叉搜索树
题目描述:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
解题思路:首先我们要理解,怎么来选点的,想要得到一个平衡二叉树,那么一定是左右结点相差深度不能超过一,因为数据是有序的,那么只要选取中间点作为根节点,以此类推,那么就一定能得到平衡二叉树,递归的去生成结点,递归有一个好处就是,可以传参数。
class Solution {
public:
TreeNode* sortedArrayToBST(vector<int>& nums) {
if(nums.size()==0)return NULL;
int mid=(nums.size()-1)/2;
TreeNode* root=new TreeNode(nums[mid]);
select_insert(0,mid-1,1,nums,root);
select_insert(mid+1,nums.size()-1,0,nums,root);
return root;
}
void select_insert(int start,int end,int l,vector<int>& nums,TreeNode* parent){
if(start>end)return;
//注意要加1,不然会导致当有两个值使用二分的话,另一个就会被丢掉,是个小细节
int mid=(end+start+1)/2;
TreeNode* cur=new TreeNode(nums[mid]);
if(l==1)parent->left=cur;
else parent->right=cur;
select_insert(start,mid-1,1,nums,cur);
select_insert(mid+1,end,0,nums,cur);
}
};