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 互不相同
题解:
由于是二叉搜索树,我们直接按照中序遍历得到的序列一定是递增有序的。
因此我们中序遍历即可,然后将两种不行的情况拉出来,然后分别指向对应应该指向的方向即可。
代码:(C语言)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int rangeSumBST(struct TreeNode* root, int low, int high){
if(root==NULL)//空时
{
return 0;
}
if(root->val>high)
//当前根节点值大于high,根据中序遍历的性质,其只能往左走,
//因为往右走的话更不符合,且val不加,因为此时val不满足
{
return rangeSumBST(root->left,low,high);
}
//同上,只可往右边走,且此时的val由于不满足所以不加
if(root->val<low)
{
return rangeSumBST(root->right,low,high);
}
return root->val + rangeSumBST(root->left,low,high) + rangeSumBST(root->right,low,high);
//因为能走到这,所以一定两个方向都是可以走的,且此时对应的val一定也满足条件,所以将满足的val以及其满足条件的方向的val全部加上即可。
}
代码:(java)
class Solution {
private int sum = 0;
public int rangeSumBST(TreeNode root, int low, int high) {
if (root == null) return 0;
rangeSumBST(root.left, low, high);
if (root.val >= low && root.val <= high) {
sum += root.val;
}
rangeSumBST(root.right, low, high);
return sum;
}
}