代码随想录day23|修剪二叉搜索树、将有序数组转化成二叉搜索树、把二叉搜索树转换成累加树

修剪二叉搜索树 

修剪一棵二叉搜索树,将节点node不在low和hig范围内的删除。们确定递归函数的参数和返回值,传入参数为节点node,最小边界low,最大边界high。

class Solution {
public:
    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;
    }
};

这道题我们采取任意一种递归顺序都可以,我写的是中左右,我们先处理一个节点,如果一个节点不在[low,high]范围内,我们返回是它修剪结束之后子树的根节点。然后我们处理左子树,对于例题的0不在[1,3]范围内,我们return的就是0的子树2->1的根节点2,我们用root->left = trimBST(root->left,low,high)来处理root的左子树,那么root->left接住的就是修剪掉0之后的子树,这样就完成了修剪操作。

将有序数组转换成二叉搜索树

 将一个有序数组转换成二叉搜索树,我们选取数组的中间数作为二叉搜索树的中节点,我们取二叉搜索树的起始节点指标为left,最后一个节点坐标为right。返回值为二叉树的中节点,中节点的左节点为子树的根坐标,中节点的右节点为子树的根左边。

class Solution {
private:
    TreeNode* traversal(vector<int>& nums,int left,int right)
    {
        if(left > right) return NULL;

        int mid = left + (right - left) / 2;
        TreeNode* node = new TreeNode(nums[mid]);//以数组中间节点为根节点
        node->left = traversal(nums,left,mid-1);
        node->right = traversal(nums,mid+1,right);
        return node;
    }
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return traversal(nums,0,nums.size()-1);
    }
};

把二叉搜索树转换成累加树

 这道题可以类比成一个数组从后往前遍历相加,我们选取两个指针用来标记当前节点的值和前一个节点的值。并且这道题是从后往前,我们采取中序的倒序来进行解题,也就是右中左。

class Solution {
private:
    int pre = 0;
    void traversal(TreeNode* node){
        if(node == NULL) return;

        traversal(node->right);//右
        node->val += pre;
        pre = node->val;
        traversal(node->left);//左
    }
public:
    TreeNode* convertBST(TreeNode* root) {
        traversal(root);
        return root;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值