[英雄星球七月集训LeetCode解题日报] 第20日 BST

最新推荐文章于 2024-11-07 23:24:07 发布

七水shuliang

最新推荐文章于 2024-11-07 23:24:07 发布

阅读量171

点赞数

分类专栏：英雄星球七月集训今天开刷leetcode 文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/liuliangcan/article/details/125884762

版权

今天开刷leetcode 同时被 2 个专栏收录

139 篇文章 3 订阅

订阅专栏

英雄星球七月集训

27 篇文章 0 订阅

订阅专栏

这篇博客分享了四道关于二叉搜索树（BST）的LeetCode解题，包括938题的BST范围和、剑指OfferII054题的所有大于等于节点的值之和，以及538和1038题的BST转换。解题策略利用BST的性质进行递归或中根遍历，有效地解决了问题。

摘要由CSDN通过智能技术生成

[英雄星球七月集训LeetCode解题日报] 第20日 BST

日报

今天两道题。

题目

一、 938. 二叉搜索树的范围和

链接: 938. 二叉搜索树的范围和

1. 题目描述

在这里插入图片描述

2. 思路分析

由于是BST，因此我们可以判断root.val的范围选择向左或向右递归。
如果val的范围介于[low,high]里，那么两边都要计算。

3. 代码实现

class Solution:
    def rangeSumBST(self, root: TreeNode, low: int, high: int) -> int:
        if not root:
            return 0
        if root.val<low:
            return self.rangeSumBST(root.right,low,high) 
        if root.val > high:
            return self.rangeSumBST(root.left,low,high)
        
        return root.val + self.rangeSumBST(root.right,low,high) + self.rangeSumBST(root.left,low,high)

二、剑指 Offer II 054. 所有大于等于节点的值之和

链接: 剑指 Offer II 054. 所有大于等于节点的值之和

1. 题目描述

在这里插入图片描述

2. 思路分析

想了好一会，甚至看了眼数据范围想遍历下来排序然后二分前缀和了。
在这个基础上想了下这是BST是有序的，按中根遍历可以省去排序；想了一下，能倒序遍历不就直接累加就行了吗。
于是发现只需要按右根左的顺序遍历累加即可。

3. 代码实现

class Solution:
    def convertBST(self, root: TreeNode) -> TreeNode:
        s = 0
        def dfs(root):
            if not root:
                return 
            nonlocal s
            dfs(root.right)
            s += root.val 
            root.val = s
            dfs(root.left)
        dfs(root)
        return root