算法复习|修剪二叉搜索树

669. 修剪二叉搜索树 - 力扣(LeetCode)

总结:这道题还是有一定难度的,自己最初想的暴力方法是先遍历一遍搜索树,得到一个vector容器,再遍历容器,当出现在范围外的点时,使用外置函数将此节点删除,比较麻烦。

递归解法:当出现在范围外的节点时,用两个节点各自接住左右两边在范围内的节点再返回这两个节点,从而达到删除当前节点的效果。

代码如下:

        if(root->val < low)
        {
            TreeNode* right = trimBST(root->right,low,high);
            return right;
        }
        if(root->val > high)
        {
            TreeNode* left = trimBST(root->left,low,high);
            return left;
        }

接着,再用当前层节点root接住返回回来的左右节点,最终达到删除效果,接着返回root节点。

全部代码如下:

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(root == nullptr)
        return nullptr;

        if(root->val < low)
        {
            TreeNode* right = trimBST(root->right,low,high);
            return right;
        }
        if(root->val > high)
        {
            TreeNode* left = trimBST(root->left,low,high);
            return left;
        }
        root->left = trimBST(root->left,low,high);
        root->right = trimBST(root->right,low,high);

        return root;
    }
};

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值