0
  • 聊天消息
  • 系统消息
  • 评论与回复
登录后你可以
  • 下载海量资料
  • 学习在线课程
  • 观看技术视频
  • 写文章/发帖/加入社区
创作中心

完善资料让更多小伙伴认识你,还能领取20积分哦,立即完善>

3天内不再提示

如何修剪二叉搜索树

算法与数据结构 来源:代码随想录 作者:程序员Carl 2021-10-11 14:16 次阅读

如果不对递归有深刻的理解,本题有点难。单纯移除一个节点那还不够,要修剪!

669. 修剪二叉搜索树

给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。

思路

相信看到这道题目大家都感觉是一道简单题(事实上leetcode上也标明是简单)。

但还真的不简单!

递归法

直接想法就是:递归处理,然后遇到root->val < low || root->val > high的时候直接return NULL,一波修改,赶紧利落。

不难写出如下代码:

classSolution{
public:
TreeNode*trimBST(TreeNode*root,intlow,inthigh){
if(root==nullptr||root->val< low || root->val>high)returnnullptr;
root->left=trimBST(root->left,low,high);
root->right=trimBST(root->right,low,high);
returnroot;
}
};

然而[1, 3]区间在二叉搜索树的中可不是单纯的节点3和左孩子节点0就决定的,还要考虑节点0的右子树

所以以上的代码是不可行的!

从图中可以看出需要重构二叉树,想想是不是本题就有点复杂了。

其实不用重构那么复杂。

在上图中我们发现节点0并不符合区间要求,那么将节点0的右孩子 节点2 直接赋给 节点3的左孩子就可以了(就是把节点0从二叉树中移除)

理解了最关键部分了我们在递归三部曲:

  • 确定递归函数的参数以及返回值

这里我们为什么需要返回值呢?

因为是要遍历整棵树,做修改,其实不需要返回值也可以,我们也可以完成修剪(其实就是从二叉树中移除节点)的操作。

但是有返回值,更方便,可以通过递归函数的返回值来移除节点。

这样的做法在二叉树:搜索树中的插入操作二叉树:搜索树中的删除操作中大家已经了解过了。

代码如下:

TreeNode*trimBST(TreeNode*root,intlow,inthigh)
  • 确定终止条件

修剪的操作并不是在终止条件上进行的,所以就是遇到空节点返回就可以了。

if(root==nullptr)returnnullptr;
  • 确定单层递归的逻辑

如果root(当前节点)的元素小于low的数值,那么应该递归右子树,并返回右子树符合条件的头结点。

代码如下:

if(root->val< low) {
    TreeNode* right = trimBST(root->right,low,high);//寻找符合区间[low,high]的节点
returnright;
}

如果root(当前节点)的元素大于high的,那么应该递归左子树,并返回左子树符合条件的头结点。

代码如下:

if(root->val>high){
TreeNode*left=trimBST(root->left,low,high);//寻找符合区间[low,high]的节点
returnleft;
}

接下来要将下一层处理完左子树的结果赋给root->left,处理完右子树的结果赋给root->right。

最后返回root节点,代码如下:

root->left=trimBST(root->left,low,high);//root->left接入符合条件的左孩子
root->right=trimBST(root->right,low,high);//root->right接入符合条件的右孩子
returnroot;

此时大家是不是还没发现这多余的节点究竟是如何从二叉树中移除的呢?

在回顾一下上面的代码,针对下图中二叉树的情况:

如下代码相当于把节点0的右孩子(节点2)返回给上一层,

if(root->val< low) {
    TreeNode* right = trimBST(root->right,low,high);//寻找符合区间[low,high]的节点
returnright;
}

然后如下代码相当于用节点3的左孩子 把下一层返回的 节点0的右孩子(节点2) 接住。

root->left=trimBST(root->left,low,high);

此时节点3的右孩子就变成了节点2,将节点0从二叉树中移除了。

最后整体代码如下:

classSolution{
public:
TreeNode*trimBST(TreeNode*root,intlow,inthigh){
if(root==nullptr)returnnullptr;
if(root->val< low) {
            TreeNode* right = trimBST(root->right,low,high);//寻找符合区间[low,high]的节点
returnright;
}
if(root->val>high){
TreeNode*left=trimBST(root->left,low,high);//寻找符合区间[low,high]的节点
returnleft;
}
root->left=trimBST(root->left,low,high);//root->left接入符合条件的左孩子
root->right=trimBST(root->right,low,high);//root->right接入符合条件的右孩子
returnroot;
}
};

精简之后代码如下:

classSolution{
public:
TreeNode*trimBST(TreeNode*root,intlow,inthigh){
if(root==nullptr)returnnullptr;
if(root->val< low) returntrimBST(root->right,low,high);
if(root->val>high)returntrimBST(root->left,low,high);
root->left=trimBST(root->left,low,high);
root->right=trimBST(root->right,low,high);
returnroot;
}
};

只看代码,其实不太好理解节点是符合移除的,这一块大家可以自己在模拟模拟!

迭代法

因为二叉搜索树的有序性,不需要使用栈模拟递归的过程。

在剪枝的时候,可以分为三步:

  • 将root移动到[L, R] 范围内,注意是左闭右闭区间
  • 剪枝左子树
  • 剪枝右子树

代码如下:

classSolution{
public:
TreeNode*trimBST(TreeNode*root,intL,intR){
if(!root)returnnullptr;

//处理头结点,让root移动到[L,R]范围内,注意是左闭右闭
while(root!=nullptr&&(root->val< L || root->val>R)){
if(root->val< L) root = root->right;//小于L往右走
elseroot=root->left;//大于R往左走
}
TreeNode*cur=root;
//此时root已经在[L,R]范围内,处理左孩子元素小于L的情况
while(cur!=nullptr){
while(cur->left&&cur->left->val< L) {
                cur->left=cur->left->right;
}
cur=cur->left;
}
cur=root;

//此时root已经在[L,R]范围内,处理右孩子大于R的情况
while(cur!=nullptr){
while(cur->right&&cur->right->val>R){
cur->right=cur->right->left;
}
cur=cur->right;
}
returnroot;
}
};

总结

修剪二叉搜索树其实并不难,但在递归法中大家可看出我费了很大的功夫来讲解如何删除节点的,这个思路其实是比较绕的。

最终的代码倒是很简洁。

如果不对递归有深刻的理解,这道题目还是有难度的!

本题我依然给出递归法和迭代法,初学者掌握递归就可以了,如果想进一步学习,就把迭代法也写一写。

其他语言版本

Java

classSolution{
publicTreeNodetrimBST(TreeNoderoot,intlow,inthigh){
if(root==null){
returnnull;
}
if(root.val< low) {
            returntrimBST(root.right,low,high);
}
if(root.val>high){
returntrimBST(root.left,low,high);
}
//root在[low,high]范围内
root.left=trimBST(root.left,low,high);
root.right=trimBST(root.right,low,high);
returnroot;
}
}

Python

classSolution:
deftrimBST(self,root:TreeNode,low:int,high:int)->TreeNode:
ifnotroot:returnroot
ifroot.val< low:
            returnself.trimBST(root.right,low,high)//寻找符合区间[low,high]的节点
ifroot.val>high:
returnself.trimBST(root.left,low,high)//寻找符合区间[low,high]的节点
root.left=self.trimBST(root.left,low,high)//root->left接入符合条件的左孩子
root.right=self.trimBST(root.right,low,high)//root->right接入符合条件的右孩子
returnroot
责任编辑:haq

声明:本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人,不代表电子发烧友网立场。文章及其配图仅供工程师学习之用,如有内容侵权或者其他违规问题,请联系本站处理。 举报投诉
  • 源代码
    +关注

    关注

    94

    文章

    2923

    浏览量

    65984
  • 二叉树
    +关注

    关注

    0

    文章

    74

    浏览量

    12210

原文标题:修剪一棵二叉搜索树

文章出处:【微信号:TheAlgorithm,微信公众号:算法与数据结构】欢迎添加关注!文章转载请注明出处。

收藏 人收藏

    评论

    相关推荐

    快速排序(2)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:58:17

    循环链表和双向链表(2)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:52:24

    广度优先搜索(2)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:48:19

    广度优先搜索(1)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:46:46

    平衡二叉树(3)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:45:06

    平衡二叉树(2)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:43:54

    平衡二叉树(1)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:42:30

    希尔排序(2)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:41:08

    二叉树二叉树的性质(3)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:09:51

    二叉树二叉树的性质(2)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:08:32

    二叉树二叉树的性质(1)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:06:44

    二叉排序树(2)(2)#数据结构

    数据函数二叉树
    未来加油dz
    发布于 :2023年09月05日 09:04:56

    NV040D语音芯片在智能毛球修剪器的应用

    毛球修剪器在不同的地方有不同的叫法,如毛球器、剃毛器、剪绒器等,是一种专门为去除衣物表面毛球而设计的小家电,除了修剪毛球之外,还可以吸出毛衣内的灰尘等杂物。而毛球修剪器存在锋利的刀片,使用不当很容易
    的头像 发表于 04-07 10:41 314次阅读
    NV040D语音芯片在智能毛球<b class='flag-5'>修剪</b>器的应用

    语音芯片在毛球修剪器上的应用—九芯电子NV040D语音ic

    毛球修剪器在不同的地方有不同的叫法,如毛球器、剃毛器、剪绒器等,是一种专门为去除衣物表面毛球而设计的小家电,除了修剪毛球之外,还可以吸出毛衣内的灰尘等杂物。 而毛球修剪器存在锋利的刀片,使用不当
    发表于 03-31 16:00 230次阅读
    语音芯片在毛球<b class='flag-5'>修剪</b>器上的应用—九芯电子NV040D语音ic

    NV040D语音芯片在毛球修剪器上的应用

    毛球修剪器在不同的地方有不同的叫法,如毛球器、剃毛器、剪绒器等,是一种专门为去除衣物表面毛球而设计的小家电,除了修剪毛球之外,还可以吸出毛衣内的灰尘等杂物。
    的头像 发表于 03-31 10:17 421次阅读