(177)LintCode 之把排序数组转换为高度最小的二叉搜索树

题目描述:给一个排序数组(从小到大),将其转换为一棵高度最小的排序二叉树。

对于一个排序好后的数组,转换成高度最小的检索树,其基本思路是:将中间值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


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值