把一個排序過的array轉成一個高度平衡二元搜尋樹中。
coding的邏輯
由於是sorted過的array 所以只要將最中間的元素當成root 然後一直不斷的遞迴下去
/**
* 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* ArrayToBST(vector<int>& nums, int begin, int end)
{
if (begin > end)
{
return nullptr;
}
int mid = (end + begin)/2;
TreeNode* root = new TreeNode(nums[mid]);
root->left = ArrayToBST(nums, begin, mid - 1);
root->right = ArrayToBST(nums, mid + 1, end);
return root;
}
TreeNode* sortedArrayToBST(vector<int>& nums)
{
return ArrayToBST(nums, 0, nums.size() - 1);
}
};