108. 将有序数组转换为二叉搜索树
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
思路
这里采用每次把vector分成两份的做法,中间的值作为根root,运用迭代left和right分别指向左右两边的值,当分到的vector为空时候就返回NULL,当nums.size()为1时leftnums、rightnums 为空,为2时将左边值计入leftnums中。
否则继续分。在if语句中定义的变量只能在if语句中使用 在else语句中定义的变量只能在else语句中使用,所以再if前定义了leftnums、rightnums ,用了作用域在if内的临时vector:temp1、temp2来赋值给leftnums、rightnums 。
/**
* 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* sortedArrayToBST(vector<int>& nums) {
if(nums.size()==0){
return NULL;
}
int n = nums.size();
int mid = n/2;
TreeNode* resT = new TreeNode(nums[mid]);
vector<int> leftnums;
vector<int> rightnums;
if(n>2){//在if语句中定义的变量只能在if语句中使用 在else语句中定义的变量只能在else语句中使用
vector<int> temp1(nums.begin(),nums.begin()+mid);
vector<int> temp2(nums.begin()+mid+1,nums.end());
leftnums = temp1;
rightnums = temp2;
}
else if(n == 2){
vector<int> temp1(nums.begin(),nums.begin()+1);
leftnums = temp1;
}
resT->left = sortedArrayToBST(leftnums);
resT->right = sortedArrayToBST(rightnums);
return resT;
}
};