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

这是一道简单题,但不要因为简单而小看他哦

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

先说代码,代码好撸,一个深度优先搜索,遍历所有节点,判断是否在low和high之间,完事。【深度优先搜索,递归】

public int RangeSumBST(TreeNode root, int low, int high)
{
    return SumLeafs(root, low, high);
}

private int SumLeafs(TreeNode node, int low, int high)
{
    var sum = 0;
    if (node.val >= low && node.val <= high)
    {
        sum += node.val;
    }

    if (node.left != null)
    {
        sum += SumLeafs(node.left, low, high);
    }
    if (node.right != null)
    {
        sum += SumLeafs(node.right, low, high);
    }

    return sum;
}

顺利通过,提交,168ms,感觉还行。 

但潜意识感觉不对劲,仔细想想,发现了问题。两个角度:

1. 都是一堆不重复的数字,是不是把数字拿下来排序,速度更快?
2. 数字拿下来理论上最快的也要先排序,排完序后一个sum范围内的就ok了。

但再仔细一想,现在都给整成二叉树了,二叉树里包含了大小关系了,为啥还要全部搜索呢?部分节点可以直接跳过去不用搜索了的。

这样就分为两种情况了:

简单说就2个case:

1. 如果当前这个节点的数字小于low了,那它的左边的子节点就没有必要再搜索了。
2. 同理,如果当前这个节点的数字大于high了,那这个节点右侧的子节点就不用再搜索了。

于是代码修改一下,增加个条件(只写了递归函数)

private int SumLeafs(TreeNode node, int low, int high)
{
    var sum = 0;
    if (node.val >= low && node.val <= high)
    {
        sum += node.val;
    }

    if (node.left != null && node.val > low)
    {
        sum += SumLeafs(node.left, low, high);
    }
    if (node.right != null && node.val < high)
    {
        sum += SumLeafs(node.right, low, high);
    }

    return sum;
}

提交,164ms 😭😭😭

这个是我比较奇怪的一个点,可能只是testcase的原因,但明显第二种方案有大量的截断和不搜索点,应该更快才对,但从提交的性能来看,并没有明显的提升。

在排除了所有可能后,我给原因归纳到leetcode平台对时间计时的不准确性,哈哈。 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值