(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


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
一个升序数组转换为高度平衡的二叉搜索树,可以采用递归的方式来实现。 具体步骤如下: 1. 找到数组的中间元素,作为二叉搜索树的根节点。 2. 将数组分成左右两个子数组,左子数组中的元素都小于根节点,右子数组中的元素都大于根节点。 3. 递归地处理左右子数组,分别将它们转换为左右子。 4. 将左右子挂在根节点下,构成一棵完整的二叉搜索树。 以下是代码示例: ```cpp #include <iostream> #include <vector> using namespace std; 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) { return buildBST(nums, 0, nums.size() - 1); } private: TreeNode* buildBST(vector<int>& nums, int left, int right) { if (left > right) { return NULL; } int mid = (left + right) / 2; TreeNode* root = new TreeNode(nums[mid]); root->left = buildBST(nums, left, mid - 1); root->right = buildBST(nums, mid + 1, right); return root; } }; int main() { vector<int> nums = {1, 2, 3, 4, 5, 6, 7}; Solution solution; TreeNode* root = solution.sortedArrayToBST(nums); // 遍历打印二叉 return 0; } ``` 在这个示例中,我们定义了一个 `Solution` 类,其中包含了一个 `sortedArrayToBST` 函数,用于将升序数组转换为高度平衡的二叉搜索树。 我们在 `sortedArrayToBST` 函数中调用了 `buildBST` 函数,用于递归地构建二叉搜索树。`buildBST` 函数的参数中,`nums` 表示原始数组, `left` 和 `right` 表示当前处理的数组区间。 在 `buildBST` 函数中,首先判断当前区间是否为空,如果为空则返回 `NULL`。然后计算出当前区间的中间位置 `mid`,将 `nums[mid]` 作为根节点。再递归地处理左右子数组,分别将它们转换为左右子。最后返回根节点,构成一棵完整的二叉搜索树。 最后,我们可以使用遍历的方式打印出生成的二叉搜索树,以验证其正确性。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值