【Leetcode】938. Range Sum of BST

原创已于 2022-10-15 12:00:32 修改 · 182 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

于 2020-04-06 09:09:03 首次发布

LC 树、分治与堆专栏收录该内容

270 篇文章

订阅专栏

本文介绍了一种解决LeetCode上二叉搜索树区间和问题的分治算法。通过判断树根节点值是否在给定范围内，递归计算左、右子树中满足条件的节点值之和。算法的时间复杂度为O(n)，空间复杂度为O(h)。

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

题目地址：

https://leetcode.com/problems/range-sum-of-bst/

给定一棵二叉搜索树，再给定一个范围 $[L, R]$ ，求这棵树中所有在这个范围里的数的和。

分治法。先判断树根是否在范围里，如果不在，那就直接求解一个子树的结果即可；否则就计算左右子树在范围中的数之和，再加上树根即可。代码如下：

class Solution {
 public:
  int rangeSumBST(TreeNode* root, int low, int high) {
    if (!root) return 0;
    int res = 0;
    if (root->val > low) res = rangeSumBST(root->left, low, high);
    if (root->val < high) res += rangeSumBST(root->right, low, high);
    if (low <= root->val && root->val <= high) res += root->val;
    return res;
  }
};

时间复杂度 $O (n)$ ，空间 $O (h)$ 。