题目描述:
给定二叉搜索树的根结点 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
解题思路:
若当前子树根节点为root,可以分为以下四种情况讨论:
1.当根节点为空,即root == NULL ,此时返回0;
2.当根节点不为空且root->val > high时,此时由于树为二叉搜索树,即树的左子树都比根节点小,右子树都比根节点大,所以此时返回左子树的范围和即可。
3.当根节点不为空 且root->val < low时,由2可知返回右子树的范围和即可。
4 root 节点的值在 [low,high] 范围内时,要返回root 节点的值、左子树的范围和、右子树的范围和这三者之和。
代码如下:
public:
int rangeSumBST(TreeNode* root, int low, int high) {
if (root == NULL) return 0;
if (root->val > high)
return rangeSumBST(root->left,low,high);
if (root->val < low)
return rangeSumBST(root->right,low,high);
else
return root->val + rangeSumBST(root->left,low,high) + rangeSumBST(root->right,low,high);
}
};
(PS:采用递归的方法固然好理解,代码业务逻辑比较清楚,但是运行的耗时也很大,在LeetCode上运行这个程序的时候花了100多ms,简直太低效了,感觉并不是最好的解决方法。)