将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
思路:其给定的数组时从小到大拍好序的,因此中间是根结点,然后左边是左子树,右边是右子树,这样二分也可以,递归也可以。
递归:
TreeNode* sortedArrayToBST(vector<int>& nums) {
if(nums.size() == 0)
return NULL;
if(nums.size() == 1)
return new TreeNode(nums[0]);
int mid = nums.size()/2;
TreeNode* r = new TreeNode(nums[mid]);
vector<int> l(nums.begin(),nums.begin()+mid);
r->left = sortedArrayToBST(l);
vector<int> ri(nums.begin()+mid+1,nums.end());
r->right = sortedArrayToBST(ri);
return r;
}
二分法:(很好理解)
TreeNode* sortedArrayToBST(vector<int>& nums) {
if(nums.size() == 0)
return NULL;
return binaryBST(nums,0,nums.size()-1);
}
TreeNode* binaryBST(vector<int>& nums,int l,int r)
{
int mid = (l + r)/2;
TreeNode* h = new TreeNode(nums[mid]);
if(l < mid)
h->left = binaryBST(nums,l,mid-1);
if(mid < r)
h->right = binaryBST(nums,mid+1,r);
return h;
}