Scala模式匹配中如何存储中间结果避免重复调用方法?
解决重复调用昂贵函数的问题
当然可以!你现在的代码确实存在重复调用expensiveTest的问题——同一个元素x会被调用两次:一次在守卫里判断是否有值,一次在右侧提取结果。如果expensiveTest是耗时操作,这会明显浪费性能,我们可以通过几种方式优化,既保留原逻辑,又只调用一次函数。
最简洁的解决方案:惰性遍历+flatMap
推荐用迭代器结合flatMap和nextOption,它和原代码的惰性行为完全一致(找到第一个符合条件的元素就停止遍历),而且代码非常简洁:
myList.iterator.flatMap(expensiveTest).nextOption()
解释:
iterator把列表转换成惰性迭代器,不会提前遍历所有元素;flatMap(expensiveTest)会对每个元素调用一次expensiveTest,将返回的Option[Int]展开——Some(value)会被保留为value,None则直接过滤掉;nextOption()取迭代器的第一个元素(如果有的话),返回Option[Int],和原代码的返回类型完全一致。
另一种写法:collectFirst + collect
如果你更习惯用collectFirst的风格,可以先收集每个元素的expensiveTest结果,再提取有效值:
myList.collectFirst { case x => expensiveTest(x) }.collect { case Some(res) => res }
解释:
collectFirst { case x => expensiveTest(x) }会遍历列表,对每个元素调用一次expensiveTest,返回第一个结果的Option[Option[Int]];- 后续的
.collect { case Some(res) => res }会把嵌套的Option展开,最终返回Option[Int]。
为什么原代码会重复调用?
原代码里的守卫case x if expensiveTest(x).isDefined会先执行一次expensiveTest(x)判断是否匹配,匹配成功后,右侧的expensiveTest(x).get又会执行一次相同的函数调用,相当于同一个元素被处理了两次。上面的方案都确保每个元素只被调用一次expensiveTest,避免了冗余计算。
内容的提问来源于stack exchange,提问作者Greg
相关产品推荐
相关产品推荐

