原题:https://leetcode-cn.com/problems/range-sum-of-bst/
使用广度优先搜索的方法,用一个队列 q 存储需要计算的节点。
每次取出队首节点时,若节点为空则跳过该节点,
否则按方法一中给出的大小关系来决定加入队列的子节点。
/**
* 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) {
queue<TreeNode*>q;//定义一个队列
//首先将根节点入队
if(root)
q.push(root);
int res = 0;
while(!q.empty())//队列非空时循环继续
{
int n = q.size();//队列的长度
for(int i = 0; i < n; i++)
{
TreeNode* t = q.front();//访问队首元素
q.pop();//队首元素出队
//注意输入格式中有空节点,所以要加一个判断
//当访问到的节点是空节点时,跳过该节点
if(t == nullptr)
{
continue;
}
//注意哦,由于是二叉搜索树,有它自己的特性
//节点的值大于high时,只需要左子树入队
if(t->val > high)
q.push(t->left);
//节点的值小于low时,只需要右子树入队
if(t->val < low)
q.push(t->right);
//节点的值在low和high之间时,需要加上该节点值以及左右子树入队
if(t->val >= low && t->val <= high)
{
res += t->val;
q.push(t->left);
q.push(t->right);
}
}
}
return res;
}
};