浅谈 二叉搜索树的范围和 问题

二叉搜索树的范围和

问题：
给定二叉搜索树的根结点 root，返回值位于范围 [low, high] 之间的所有结点的值的和。

思路：
法一：深度优先搜索（dfs）

1. 如果没有根节点返回 0；
2. 遍历节点值，如果比 low 小则去掉左子树，比 high 大则去掉右子树；
3. 返回符合条件的所有节点值。

法二：广度优先搜索（bfs）

1. 用 队列 来实现树的广度优先搜索
   使用广度优先搜索的方法，用一个队列 q 存储需要计算的节点。每次取出队首节点时，若节点为空则跳过该节点，否则按方法一中给出的大小关系来决定加入队列的子节点。

// dfs
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int rangeSumBST(TreeNode* root, int low, int high) {
        if(!root) return 0;
        if(root->val < low) {
            return rangeSumBST(root->right, low, high);
        }
        if(root->val > high) {
            return rangeSumBST(root->left, low, high);
        }
        return root->val + rangeSumBST(root->right, low, high) + rangeSumBST(root->left, low, high);
    }
};

• 时间复杂度：O(n)
• 空间复杂度：O(n)

// bfs
class Solution {
public:
    int rangeSumBST(TreeNode *root, int low, int high) {
        int sum = 0;
        queue<TreeNode*> q({root});
        while (!q.empty()) {
            auto node = q.front();
            q.pop();
            if (node == nullptr) {
                continue;
            }
            if (node->val > high) {
                q.push(node->left);
            } else if (node->val < low) {
                q.push(node->right);
            } else {
                sum += node->val;
                q.push(node->left);
                q.push(node->right);
            }
        }
        return sum;
    }
};

• 时间复杂度：O(n)
• 空间复杂度：O(n)