代码随想录训练营第二十三天| ● 669. 修剪二叉搜索树 ● 108.将有序数组转换为二叉搜索树 ● 538.把二叉搜索树转换为累加树

669. 修剪二叉搜索树 

题目链接/文章讲解:  代码随想录
TreeNode* trimBST(TreeNode* root, int low, int high) {
        if (root == NULL) return NULL;
        if (root->val < low) {
            return trimBST(root->right, low, high);
        }
        if (root->val > high) {
            return trimBST(root->left, low, high);
        }
        root->left = trimBST(root->left, low, high);
        root->right = trimBST(root->right, low, high);
        return root;
    }

修建二叉搜索树,需要利用二叉搜索树的特性,左子树小于父节点,右子树大于父节点。因此当前节点小于目标区域时,应该返回右子树(有可能满足条件);大于时应该返回左子树。

最后递归的求解左右子树即可完成。 

108.将有序数组转换为二叉搜索树

TreeNode* sortedArrayToBST(vector<int>& nums) {
        int size = nums.size();
        if (size == 0) return 0;
        TreeNode* node = new TreeNode(nums[size/2]);
        vector<int> left(nums.begin(), nums.begin()+size/2);
        node->left = sortedArrayToBST(left);
        vector<int> right(nums.begin()+size/2+1, nums.end());
        node->right = sortedArrayToBST(right);

        return node;
    }

 自己的递归做法,由于是平衡二叉搜索树,所以数组中间的元素肯定是二叉树的根节点(这样才能保持平衡),因此,利用数组中间元素来构建二叉树。因为要从上向下构建,要采用中序遍历。首先通过中间元素来当做根节点,然后使用左侧数组来构建左子树,使用右侧数组来构建右子树,这样就能构建出二叉搜索数了。

class Solution {
public:
    int sum = 0;
    TreeNode* convertBST(TreeNode* root) {
        if (root == NULL) return NULL;
        convertBST(root->right);
        sum += root->val;
        root->val = sum;
        convertBST(root->left);
        return root;
    }
};

我使用了一个全局变量来记录累加的和,然后进行右中左的中序遍历,从最右下的节点一路累加,直至加完整棵树即可得到累加树。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值