LeetCode-938:二叉搜索树的范围和

给定二叉搜索树的根结点 root,返回 L 和 R(含)之间的所有结点的值的和。

二叉搜索树保证具有唯一的值。

示例 1:
输入:root = [10,5,15,3,7,null,18], L = 7, R = 15
输出:32

示例 2:
输入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出:23

提示:
树中的结点数量最多为 10000 个。
最终的答案保证小于 2^31。

代码实现
C++:

题目的意思是要找出所有值在区间[L,R]内的节点,求它们的和。
所以核心问题就是查找。在这里插入图片描述

LeetCode基础算法题第145篇:BST的范围总和
如上图,根据二叉搜索树,当L=6,R=15时,红框内就是我们要找的所有节点。

我们会发现,如果节点值小于等于6,没必要再查找它的左子树;同样的如果节点值大于等于15,没必要再查找它的右子树。

换句话说:

如果节点大于等于L,就需要查找左子树;
如果节点小于等于R,就需要查找右子树;
对于同时满足上述两个条件的节点,左右子树都需要查找。所以上述两个查找条件,可以作为并列的两个独立条件。

此外我们还需要另外的两个必要的判断:

当节点是空节点的时候,应该返回0,因为查找的时候会查找到叶子节点;
只有节点值在区间[L,R]内才统计它的值;

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int rangeSumBST(TreeNode* root, int L, int R) {
        if(root == NULL){
            return 0;
        }
        int sum = 0;
        if(root->val >= L && root->val <= R){
            sum = root->val;
        }
        //访问左树
        if(root->val >= L){
            sum += rangeSumBST(root->left, L, R);
        }
        //访问又树
        if(root->val <= R){
            sum += rangeSumBST(root->right, L, R);
        }
        return sum;
    }
};

Python:

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def rangeSumBST(self, root, L, R):
        """
        :type root: TreeNode
        :type L: int
        :type R: int
        :rtype: int
        """
        if not root:
            return 0
        sum = 0
        if L <= root.val <= R:
            sum = root.val
        # 访问左子树
        if root.val >= L:
            sum += self.rangeSumBST(root.left, L, R)
        # 访问右子树
        if root.val <= R:
            sum += self.rangeSumBST(root.right, L, R)
        
        return sum
        
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值