题解 | #农场牛的最佳观赏区间# 二叉搜索树

知识点

二叉搜索树

思路

总的思路是遍历二叉树的每一个点，累加在范围内的值，得到答案。但是由于二叉搜索树的左子树的值一定小于根节点小于右子树的值，所以当根节点的值大于low才遍历左子树，当根节点的值小于high才遍历右子树，可以优化常数。最坏复杂度不会变。

时间复杂度

最坏情况下每一个点都在范围内，遍历一遍二叉树，时间复杂度为。

AC code（C++）

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param low int整型 
     * @param high int整型 
     * @return int整型
     */
    int rangeSumBST(TreeNode* root, int low, int high) {
        int res = 0;
        function<void(TreeNode*)> dfs = [&](TreeNode* root) {
            if (!root) return;
            if (root->val > low) dfs(root->left);
            if (root->val >= low and root->val <= high) res += root->val;
            if (root->val < high) dfs(root->right);
        };
        dfs(root);
        return res;
    }
};