Leetcode 938.二叉搜索树的范围和Java&Python

最新推荐文章于 2023-02-24 11:54:51 发布

江城的程序员大叔

最新推荐文章于 2023-02-24 11:54:51 发布

阅读量773

点赞数

CC 4.0 BY-SA版权

分类专栏：你好，leetcode 文章标签： Leetcode 938.二叉搜索树的范围和 Java Python

本文链接：https://blog.youkuaiyun.com/sinat_33150417/article/details/84326854

你好，leetcode 专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找指定区间内所有节点值之和的方法。通过利用二叉搜索树的特性，可以高效地遍历并计算出结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定二叉搜索树的根结点 root，返回 L 和 R（含）之间的所有结点的值的和。

二叉搜索树保证具有唯一的值。

示例 1：

输入：root = [10,5,15,3,7,null,18], L = 7, R = 15
输出：32

示例 2：

输入：root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出：23

提示：

树中的结点数量最多为 10000 个。
最终的答案保证小于 2^31。

题目涉及到二叉搜索树而不是普通的二叉树，那么在考虑问题的时候，就一定要联想到二叉搜索树的性质。根据题目描述，首先想到的就是遍历整棵树，找到给定区间里的值求和即可，这是普通二叉树的做法，既然是二叉搜索树，那么当我们找到一个节点比L小的时候，那么以这个节点为根的子树就不需要遍历了，因为二叉搜索树的左孩子（子树）小于根，右侧的R也是同样的道理，下面是AC代码：

Java：

public int rangeSumBST(TreeNode root, int L, int R) {
        if (root == null) {
            return 0;
        }
        
        if (root.val < L) {
            return rangeSumBST(root.right, L, R);
        }
        
        if (root.val > R) {
            return rangeSumBST(root.left, L, R);
        }
        
        return root.val + rangeSumBST(root.right, L, R) + rangeSumBST(root.left, L, R);
    }

Python：

def rangeSumBST(self, root, L, R):
        """
        :type root: TreeNode
        :type L: int
        :type R: int
        :rtype: int
        """
        if root is None: return 0

        if root.val < L:
            return self.rangeSumBST(root.right, L, R)

        if root.val > R:
            return self.rangeSumBST(root.left, L, R)

        return root.val + self.rangeSumBST(root.left, L, R) + self.rangeSumBST(root.right, L, R)