给定二叉搜索树的根结点 root,返回 L 和 R(含)之间的所有结点的值的和。
二叉搜索树保证具有唯一的值。
示例 1:
输入:root = [10,5,15,3,7,null,18], L = 7, R = 15
输出:32
示例 2:
输入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出:23
提示:
树中的结点数量最多为 10000 个。
最终的答案保证小于 2^31。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-of-bst
思路:用递归结构遍历二叉搜索树,累加在 L 和 R 之间的节点值。
【二叉搜索树】:节点值按照大小顺序排序构成的树
一开始设置了count作为全局变量,想要通过全局变量累加每次符合条件的节点数值,但是!leetcode在用全局变量提交的时候居然会出现与测试不一致的错误!最后思来想去通过函数本身返回的数值一次次累加,才能够正确返回。
int rangeSumBST(struct TreeNode* root, int L, int R){
int count = 0;
if (root->left != NULL)//不能root->left->vel >= L
count += rangeSumBST(root->left, L, R);
if (root->val <= R && root->val >= L)
count += root->val;
//printf("%d\t",root->val);
if (root->right != NULL)
count += rangeSumBST(root->right, L, R);
return count;
}
【关于count】
在一开始设定count变量并初始化为0,在二叉树中序遍历的基础上,遍历的同时因为函数同时返回本次计算的count,因此通过一个加法来拿到前次递归得到的count数据,存进count内。
在其后,加上本次符合要求的节点值,若节点不符合,count将传递以前节点计算出来的count值,并不加任何数值,从而实现递归之中数据传递。
【关于剪枝】
二叉搜索树往左依次变小,往右依次变大,但左子树最大值不超过根节点,右子树最小值不小于根节点,因此在遍历时不能贸然剪枝,会出现一层一层变大或变小情况,当遍历到一个点时并不知道左子树或者右子树是否会出现符合条件的点。
总之是我太菜了想不出来有什么剪枝的好方法。
【成绩】
116 ms(90%) 43.8 MB(100%)