如何在 scala.collection.mutable.HashMap 中使用 getOrElseUpdate？

Question

示例代码计算给定输入文件中每个单词的出现次数：

object Main {
  def main(args: Array[String]) {
    val counts = new scala.collection.mutable.HashMap[String, Int]
    val in = new Scanner(new File("input.txt"))

    while (in.hasNext()) {
      val s: String = in.next()
      counts(s) = counts.getOrElse(s, 0) + 1 // Here!
    }

    print(counts)
  }
}

可以使用getOrElseUpdate 方法重写带有注释突出显示的行吗？

附：我只是在“不耐烦的 Scala”的第 4 部分，所以请不要现在教我函数式 Scala，我敢肯定，这里可以更漂亮。

谢谢。

Answer 1

如果您查看doc，您会看到下一个：

如果给定的键已经在这个映射中，返回相关的值。 否则，从给定的表达式 op 计算值，用 key in 存储 映射并返回该值。

，不过还是要修改地图，所以getOrElseUpdate在这里没用。

你可以定义default的值，如果key不存在就会返回。并用下一种方式：

import scala.collection.mutable.HashMap
object Main {
  def main(args: Array[String]) {
    val counts = HashMap[String, Int]().withDefaultValue(0)
    val in = new Scanner(new File("input.txt"))

    while (in.hasNext()) {
      val s: String = in.next()
      counts(s) += 1
    }

    print(counts)
  }
}