这是一道简单题,但不要因为简单而小看他哦
给定二叉搜索树的根结点
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平台对时间计时的不准确性,哈哈。