题目描述:给一个排序数组(从小到大),将其转换为一棵高度最小的排序二叉树。
对于一个排序好后的数组,转换成高度最小的检索树,其基本思路是:将中间值A[mid]作为树根,比中间值小的放在左子树,比中间值大的放在右子树。这是一个典型的递归操作。
考虑到Lintcode中给出的函数只给了一个参数,我们需要构造新的vector来分别存放A[mid]左边的值和A[mid]右边的值。代码如下:
class Solution {
public:
/*
* @param A: an integer array
* @return: A tree node
*/
vector<int>B;
TreeNode * sortedArrayToBST(vector<int> &A) {
// write your code here
if(A.empty())
return NULL;
vector<int>::iterator iter=A.begin();
vector<int>::iterator iter_end=A.end();
int size=A.size();
int mid=(size-1)/2;
TreeNode * root=new TreeNode(A[mid]);
B.assign(iter,iter+mid);
root->left=sortedArrayToBST(B);
B.assign(iter+mid+1,iter_end);
root->right=sortedArrayToBST(B);
}
};
仔细看上述实现方式,我们发现,每次递归调用时,都要进行一个B.assign()的赋值操作,这个操作代价较高,所以我又想了另一个方式:对左右子树调用能用下标划分向量的函数。(这边传递向量A的引用,而不是A本身,可以降低一半的代价)
因为考虑到向量是有序的,所以用下标查找的复杂度(θ(1))会远低于使用指针查找的复杂度(θ(n)),所以用下标作为函数调用的参数。如下:
class Solution {
public:
/*
* @param A: an integer array
* @return: A tree node
*/
TreeNode * mySAtoBST(vector<int>&A,int begin,int end){
if(begin>end)
return NULL;
int mid=(end+begin)/2;
TreeNode * root=new TreeNode(A[mid]);
root->left=mySAtoBST(A,begin,mid-1);
root->right=mySAtoBST(A,mid+1,end);
return root;
}
TreeNode * sortedArrayToBST(vector<int> &A){
if(A.empty())
return NULL;
int size=A.size();
int mid=(size-1)/2;
TreeNode * root=new TreeNode(A[mid]);
root->left=mySAtoBST(A,0,mid-1);
root->right=mySAtoBST(A,mid+1,size-1);
return root;
}
};
————————————————————————————————————————————————————
我曾在mid的取值这边纠结了一会,因为自己一直在“对边界的判断”这方面不太擅长。这次实现这个算法之后,记住了一个显而易见但我老是没有考虑到的结论:对下标begin,end,和容器容量size,有这样一个关系:begin+end=size-1