【Leetcode】938.二叉搜索树的范围和(Java+JavaScript)

一、题目描述

二叉搜索树的范围和

给定二叉搜索树的根结点 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

提示:

树中节点数目在范围 [1, 2 * 104] 内
1 <= Node.val <= 105
1 <= low <= high <= 105
所有 Node.val 互不相同

二、题目解法

1、中序遍历

二叉搜索树中序遍历可以得到一个递增的集合,根据这个特点,就可以计算出符合规则的节点值和

	 public int rangeSumBST(TreeNode root, int low, int high) {
        if (root == null)
            return 0;
        int current = root.val <= high && root.val >= low ? root.val : 0;
        return rangeSumBST(root.left, low, high) + current + rangeSumBST(root.right, low, high);
    }

2、中序遍历 + 剪枝

上个版本中,递归遍历了二叉搜索树中的所有节点,所以,有没有方式可以优化呢,只遍历符合条件的节点值?

只要在递归过程中,判断节点值是否满足要查询的节点值的范围之内即可,这样,就可以去掉不符合题意的节点值的递归

java版:

	public int rangeSumBST(TreeNode root, int low, int high) {
		if (root == null)
            return 0;
        var sum = 0;
        if (root.val > low)
            sum += rangeSumBST(root.left, low, high);
        sum += root.val <= high && root.val >= low ? root.val : 0;
        if (root.val < high)
            sum += rangeSumBST(root.right, low, high);
        return sum;
   }

javascript版:

var rangeSumBST = function(root, low, high) {
    if (!root)
        return 0
    let sum = 0
    if (root.val >= low)
        sum += rangeSumBST(root.left, low, high)
    sum += root.val <= high && root.val >= low ? root.val : 0
    if (root.val <= high)
        sum += rangeSumBST(root.right, low, high)
    return sum
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值