1、描述
给定二叉搜索树的根结点 root,返回值位于范围 [low, high] 之间的所有结点的值的和。
2、关键字
二叉搜索树、区间
3、思路
dfs,先根遍历,一层一层比较当前值,递归,先写当前val,再判断左右子树
4、notes
递归函数,返回值写成void,然后通过一个全局变量保存结果
2、给出的主函数,不能重定义全局变量,不然不是同一个变量,没有结果
5、复杂度
时间:O(N)节点个数
空间:O(H)高度
6、code
/**
* 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) {}
* };
感觉一般递归函数都是void类型的,然后搞一个全局变量,用来记录结果,
*/
class Solution {
public:
int res;
void dfs(TreeNode * root,int l,int r){ //
if(root != nullptr){
if(root->val <= r && root->val >= l)
res+= root->val;
if(l <= root->val) // 等号用不用都可以通过,就看二叉搜索树是否有重复的数字了
dfs(root->left,l,r);
if(r >= root->val)
dfs(root->right,l,r);
}
}
int rangeSumBST(TreeNode* root, int low, int high) {
if(!root) return 0;
// int res = 0; // 这里如果局部重定义,结果就不对了!!!
res = 0;
dfs(root,low,high);
return res;
}
};