【Leetcode】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 互不相同
思路
按深度优先搜索的顺序计算范围和。记当前子树根节点为 root，分以下四种情况讨论：
- 当根节点为空时，应直接返回0。
- 若根节点的值大于high，因为二叉搜索树右子树上的所有节点值均大于根节点的值，故只需考虑左子树的范围和。
- 若根节点的值小于low，因为二叉搜索树左子树上的所有节点值均小于根节点的值，故只需考虑右子树的范围和。
- 当根节点的值在[low, high]范围内时，应返回根节点的值、左子树的范围和、右子树的范围和的总和。
此时应返回 root 节点的值、左子树的范围和、右子树的范围和这三者之和。
代码
class Solution {
public:int rangeSumBST(TreeNode* root, int low, int high) {int ans=0;if(root==nullptr)return 0;if(root->val > high)return rangeSumBST(root->left,low,high);else if(root->val < low)return rangeSumBST(root->right,low,high);else return root->val + rangeSumBST(root->left, low, high) + rangeSumBST(root->right, low, high);}
};
结论
