LeetCode 938. 二叉搜索树的范围和 C++

题目描述

给定二叉搜索树的根结点 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 互不相同

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-sum-of-bst

解题思路

比较无脑的解法：不考虑二叉搜索树的特性，直接前序遍历扫一遍，如果当前扫到的结点在[low,high]之间，就加入我们的计数和res。
比较优秀的解法应该是利用好二叉搜索树的特性去扫。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void preOrder(TreeNode* root,int low,int high,int &res){
        if(root==NULL){//如果当前扫到了空结点，直接返回
            return ;
        }
        if(root->val>=low&&root->val<=high){//如果当前结点符合条件，就加入res
            res+=root->val;
        }
        preOrder(root->left, low, high, res);//前序遍历左孩子
        preOrder(root->right, low, high, res);//前序遍历右孩子
    }
    int rangeSumBST(TreeNode* root, int low, int high) {
        int res=0;//用来返回和
        preOrder(root,low,high,res);//前序遍历root，并且把结果返回给res
        return res;//返回res
    }
};