题目:
Given an array where elements are sorted in ascending order, convert it to a height balanced BST.
题意:
给一个排好序的数组,要求把数组转化为一个深度平均的搜索二叉树。
对于二叉树,父节点一定 大于左子节点,小于右子节点,所以对应数组而言,父节点应为中间值,比它小的数放在左边,大的数放在右边,依次递归下去。
代码如下:
public:
TreeNode * newBST(vector<int>nums, int l, int r) {
if (l > r) return NULL;
int mid = (l + r) / 2;
TreeNode * node = new TreeNode(nums[mid]);
node->left = newBST(nums, l, mid-1);
node->right = newBST(nums, mid+1, r);
return node;
}
TreeNode* sortedArrayToBST(vector<int>& nums) {
return newBST(nums, 0, nums.size()-1);
}
};