leetcode 938.二叉搜索树的范围和

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 * 104] 内
1 <= Node.val <= 105
1 <= low <= high <= 105
所有 Node.val 互不相同

知识点&算法

给定二叉搜索树，求区间和。说实话一开始想到的解法是线段树，但是在树结构原本存在的情况下再建树好像空间开销有点大。
于是想到用前缀和记录BST中序遍历下的和，然后搜索符合给定区间的范围，再对前缀和进行差运算。

既然都线性遍历节点了为什么不干脆在遍历过程中做加运算呢…

题解

class Solution {
public:
    int preSum[20005],nodes[20005],cnt = 1;
    void recur(TreeNode* node){
        if(node == nullptr) return;
        recur(node->left);
        nodes[cnt] = node->val;
        preSum[cnt++] = node->val + preSum[cnt-1];
        recur(node->right);
        return;
    }
    int rangeSumBST(TreeNode* root, int low, int high) {
        int res = 0,l = 0x3f3f3f3f,r = -1;
        recur(root);
        for(int i = 1 ; i < cnt ; ++i){
            if(nodes[i] >= low) l = min(l,i);
            if(nodes[i] >= high){
                r = i;
                break;
            }
        }
        return preSum[r] - preSum[l-1];
    }
};

直接加：

class Solution {
public:
    int sum = 0,low,high;
    void recur(TreeNode* node){
        if(node == nullptr) return;
        recur(node->left);
        if(node->val >= low && node->val <= high) sum += node->val;
        recur(node->right);
        return;
    }
    int rangeSumBST(TreeNode* root, int low, int high) {
        this->low = low;
        this->high = high;
        recur(root);
        return sum;
    }
};