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

写代码的第十八天,二叉树最后一天!!!加油呀!!!
现在的重点还是在递归上,迭代我先放一放了,二刷的时候再重点看迭代。
今天居然没有写错的代码,稀奇了,整理思路还是有用。

669. 修剪二叉搜索树

思路(递归)

找到需要删除的结点,如果该结点的值小于最小范围值,那么递归遍历其右子树,把右子树中符合范围的子树保留返回给待删除结点的父结点;如果该结点的值大于最大范围值,那么递归其左子树,把左子树中符合范围的子树保留返回给待删除结点的父结点;现在处理完根结点了,然后就需要处理左右子树,那么就需要递归的从根结点开始遍历,先遍历左子树,在遍历右子树。
解决问题1:参数和返回值?参数需要当前树的根结点以及修剪的范围;返回值是新树的根结点。
解决问题2:终止条件?我们要遍历完整颗树,所以终止条件是if root= =None,返回None。
解决问题3:单次递归逻辑?按照中左右的方式遍历,首先处理中间结点,如果root.val<left,那么trimBST(root.right,left,right),然后return 上面的式子;如果root.val<right,那么trimBST(root.left,left,right),然后return上面的式子。接下来处理左右子树,还是trimBST(root.left,left,right)和trimBST(root.right,left,right)。
正确代码

class Solution:
    def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
        if root == None:
            return None
        if root.val < low:
            right = self.trimBST(root.right,low,high)
            return right
        if root.val > high:
            left = self.trimBST(root.left,low,high)
            return left
        root.left = self.trimBST(root.left,low,high)
        root.right = self.trimBST(root.right,low,high)
        return root

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

思路(递归)

这个题和之前的中后序构造二叉树这类题很相似。采用的是一种方法,都是找到结点然后切割左右两部分,左边递归,右边递归。
解决问题1:参数和返回值?参数需要当前树的根结点数组的左右范围;返回值是新树的根结点。有点像二分查找诶
解决问题2:终止条件?我们是根据有序数组来构建二叉树,所以终止条件和树中的结点没关系,应该是当数组中无数据了这时候终止。那条件应该怎么写呢?我们在函数中传入了数组的左和右然后找mid中结点,两边递归,这个时候左右就很像二分查找中的left指针和right指针,回想一下在二分查找中循环停止条件是left<right,这个条件产生的原因是我们采用的左闭右开的区间,所以本题中也可以用左闭右开的情况,当left大于right的时候停止递归。
解决问题3:单次递归逻辑?首先找到根结点,也就是数组中的中间结点,当数组长度是奇数的时候直接用数组中间的值,当数组长度是偶数的时候我这里设置根结点为(left+right)/2这个位置的值。然后递归遍历左子树和右子树。
正确代码

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
        return self.traversel(nums,0,len(nums))

    def traversel(self,nums,left,right):
        if left >= right:
            return None
        mid = (left + right) // 2
        root = TreeNode(nums[mid])
        root.left = self.traversel(nums,left,mid)
        root.right = self.traversel(nums,mid+1,right)
        return root

538.把二叉搜索树转换为累加树

思路(递归)

这个题要求的是累加树,把比这个结点值大的那些都加起来作为当前结点的值,这是一颗二叉搜索树,所以本身就是有序的,是从小到大的,按照左中右顺序,但是现在要从大到小的累加,那么我们可以反向一下,右中左这个顺序就是从大到小了。每一次都需要累加一个结点所以我们可以设置两个指针,一个遍历整颗树,一个在这个指针之前pre用来记录前一个结点的值,这样每次就不需要遍历整颗树重新相加了,只需要每次当前结点加pre就可以了。
解决问题1:参数和返回值?参数是跟结点,返回值也是新的树根结点。
解决问题2:终止条件?当到叶子结点的时候也就是root==空的时候停止。
解决问题3:单次递归逻辑?按照右中左的逻辑遍历,设置全局变量pre=0,为了累加计数,设置cur指针从最右侧开始,每次累加更新值
正确代码

class Solution:
    def __init__(self):
        self.pre = 0
        
    def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        root = self.traversal(root)
        return root
        
    def traversal(self,cur):
        if cur == None:
            return None
        cur.right = self.traversal(cur.right)
        cur.val += self.pre
        self.pre = cur.val
        cur.left = self.traversal(cur.left)
        return cur
  • 5
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
代码随想录算法训练是一个优质的学习和讨论平台,提供了丰富的算法训练内容和讨论交流机会。在训练中,学员们可以通过观看视频讲解来学习算法知识,并根据讲解内容进行刷题练习。此外,训练还提供了刷题建议,例如先看视频、了解自己所使用的编程语言、使用日志等方法来提高刷题效果和语言掌握程度。 训练中的讨论内容非常丰富,涵盖了各种算法知识点和解题方法。例如,在第14天的训练中,讲解了二的理论基础、递归遍历、迭代遍历和统一遍历的内容。此外,在讨论中还分享了相关的博客文章和配图,帮助学员更好地理解和掌握二的遍历方法。 训练还提供了每日的讨论知识点,例如在第15天的讨论中,介绍了层序遍历的方法和使用队列来模拟一层一层遍历的效果。在第16天的讨论中,重点讨论了如何进行调试(debug)的方法,认为掌握调试技巧可以帮助学员更好地解决问题和写出正确的算法代码。 总之,代码随想录算法训练是一个提供优质学习和讨论环境的平台,可以帮助学员系统地学习算法知识,并提供了丰富的讨论内容和刷题建议来提高算法编程能力。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [代码随想录算法训练每日精华](https://blog.csdn.net/weixin_38556197/article/details/128462133)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值