把一個排序過的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); } };
沒有留言 :
張貼留言