Scala：使用具有默认值的 HashMap

Question

我有一个可变的 HashMap，想像使用默认字典一样使用它。显而易见的方法似乎是使用 getOrElse 并每次都提供默认值作为第二个值。然而，这在我的用例中似乎有点不雅，因为默认值没有改变。

var x = HashMap(1 -> "b", 2 -> "a", 3 -> "c")

println(x.getOrElse(4, "_")
println(x.getOrElse(5, "_"))
// And so on...
println(x.getOrElse(10, "_"))

有什么方法可以创建一个 HashMap（或类似的类），以便尝试访问未定义的键返回一个在创建 HashMap 时设置的默认值？我注意到 HashMap.default 只是设置为抛出异常，但我想知道这是否可以更改...

Answer 1

哇，在我在这里发布我的最后一个答案一年后，我碰巧访问了这个帖子。 :-)

Scala 2.9.1。 mutable.Map 带有 withDefaultValue 方法。 REPL 会话：

scala> import collection.mutable
import collection.mutable

scala> mutable.Map[Int, String]().withDefaultValue("")
res18: scala.collection.mutable.Map[Int,String] = Map()

scala> res18(3)
res19: String = ""

Answer 2

scala> val x = HashMap(1 -> "b", 2 -> "a", 3 -> "c").withDefaultValue("-")
x: scala.collection.immutable.Map[Int,java.lang.String] = Map((1,b), (2,a), (3,c))

scala> x(3)
res0: java.lang.String = c

scala> x(5)
res1: java.lang.String = -

编辑：

对于mutable.HashMap，您可以执行以下操作：

scala> import collection.mutable
import collection.mutable

scala> val x = new mutable.HashMap[Int, String] {
     |   override def apply(key: Int) = super.get(key) getOrElse "-"
     | }
x: scala.collection.mutable.HashMap[Int,String] = Map()

scala> x += (1 -> "a", 2 -> "b", 3 -> "c")
res9: x.type = Map((2,b), (1,a), (3,c))

scala> x(2)
res10: String = b

scala> x(4)
res11: String = -

可能有更好的方法来做到这一点。等待其他人回应。

Answer 3

试试这个：

import collection.mutable.HashMap
val x = new HashMap[Int,String]()  { override def default(key:Int) = "-" }
x += (1 -> "b", 2 -> "a", 3 -> "c")

然后：

scala> x(1)
res7: String = b

scala> x(2)
res8: String = a

scala> x(3)
res9: String = c

scala> x(4)
res10: String = -

Answer 4

我更像是一个 java 人...但是如果 getOrElse 不是最终的，你为什么不扩展 HasMap 并提供这样的东西：

override def getOrElse(k: Int, default: String) = {
  return super.getOrElse(k,"_")
}

注意：语法可能搞砸了，但希望你能明白这一点