【Leetcode】938. 二叉搜索树的范围和

题目

题目链接
给定二叉搜索树的根结点 root，返回值位于范围 [low, high] 之间的所有结点的值的和。

示例 1：

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

示例 2：

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

提示：

• 树中节点数目在范围 [1, 2 * 10⁴] 内
• 1 <= Node.val <= 10⁵
• 1 <= low <= high <= 10⁵
• 所有 Node.val 互不相同

思路

按深度优先搜索的顺序计算范围和。记当前子树根节点为 root，分以下四种情况讨论：

1. 当根节点为空时，应直接返回0。
2. 若根节点的值大于high，因为二叉搜索树右子树上的所有节点值均大于根节点的值，故只需考虑左子树的范围和。
3. 若根节点的值小于low，因为二叉搜索树左子树上的所有节点值均小于根节点的值，故只需考虑右子树的范围和。
4. 当根节点的值在[low, high]范围内时，应返回根节点的值、左子树的范围和、右子树的范围和的总和。

此时应返回 root 节点的值、左子树的范围和、右子树的范围和这三者之和。

代码

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

结论