二叉搜索树的范围和
二叉搜索树的范围和
题目:
给定二叉搜索树的根结点 root
,返回 L
和 R
(含)之间的所有结点的值的和。
二叉搜索树保证具有唯一的值。
思路:
我们对树进行深度优先搜索,对于当前节点 node,如果 node.val
小于等于 L,那么只需要继续搜索它的右子树;如果 node.val
大于等于 R,那么只需要继续搜索它的左子树;如果 node.val
在区间 (L, R) 中,则需要搜索它的所有子树。
代码:
class Solution {
int ans;
public int rangeSumBST(TreeNode root, int L, int R) {
ans = 0;
dfs(root, L, R);
return ans;
}
public void dfs(TreeNode node, int L, int R) {
if (node != null) {
if (L <= node.val && node.val <= R)
ans += node.val;
if (L < node.val)
dfs(node.left, L, R);
if (node.val < R)
dfs(node.right, L, R);
}
}
}