题
给定一个有序整数数组,元素各不相同且按升序排列,编写一个算法,创建一棵高度最小的二叉搜索树。
示例:
给定有序数组: [-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;
}
TreeNode *ans;
sort(nums.begin(), nums.end());
doIt(ans, nums);
return ans;
}
void doIt(TreeNode* &node, vector<int> nums)
{
if (nums.size() == 0)
{
return;
}
int middle = (nums.size() - 1) / 2;
node = new TreeNode(nums[middle]);
vector<int>l,r;
for (int i = 0; i < nums.size(); i++)
{
if (i == middle)
{
continue;
}
if (i<middle)
{
l.push_back(nums[i]);
}
else
{
r.push_back(nums[i]);
}
}
doIt(node->left, l);
doIt(node->right, r);
}
};