Scala vals vs vars

Question

我对Scala很新,但我想知道解决这个问题的首选方法是什么.假设我有一个项目列表,我想知道检查项目的总数.我可以这样做:

val total = items.filter(_.itemType == CHECK).map(._amount).sum

这将给我我需要的东西,即不可变变量中所有检查的总和.但它看起来像是3次迭代.一旦过滤检查,再次映射金额,然后总和.另一种方法是做一些事情:

var total = new BigDecimal(0)
for (
    item <- items
    if item.itemType == CHECK
) total += item.amount

这给了我相同的结果,但是1次迭代和一个可变变量看起来也很好.但是,如果我想提取更多信息,比如支票总数,那就需要更多的计数器或可变变量,但我不必再次遍历列表.似乎不是实现我需要的"功能"方式.

var numOfChecks = 0
var total = new BigDecimal(0)
items.foreach { item =>
    if (item.itemType == CHECK) {
        numOfChecks += 1
        total += item.amount
    }
}

因此,如果您发现自己需要在列表中使用一堆计数器或总计,则首选保留可变变量或不担心它会执行以下操作:

val checks = items.filter(_.itemType == CHECK)
val total = checks.map(_.amount).sum
return (checks.size, total)

这似乎更容易阅读,只能使用 vals

Answer 1

在一次迭代中解决问题的另一种方法是使用视图或迭代器:

items.iterator.filter(_.itemType == CHECK).map(._amount).sum

要么

items.view.filter(_.itemType == CHECK).map(._amount).sum

这样,表达式的评估被推迟到调用sum.

如果您的项目是案例类,您也可以这样写:

items.iterator collect { case Item(amount, CHECK) => amount } sum

Answer 2

我发现说"三次迭代"有点误导 - 毕竟,每次迭代都比一次迭代完成的工作要少.所以它不会自动跟随迭代三次将比迭代一次更长.

创建临时对象,现在这是一个问题,因为你将打击内存(即使是缓存),这不是单次迭代的情况.在这些情况view下,即使它添加了更多方法调用来执行相同的工作,也会有所帮助.希望JVM能够优化它.有关视图的更多信息,请参阅Moritz的答案.

Answer 3

您可以使用foldLeft:

(0 /: items) ((total, item) => 
   if(item.itemType == CHECK) 
      total + item.amount 
   else 
      total
)

以下代码将返回一个元组(支票数 - >金额总和):

((0, 0) /: items) ((total, item) => 
   if(item.itemType == CHECK) 
      (total._1 + 1, total._2 + item.amount) 
   else 
      total
)