scala - 如何更新 mutable.Map 中的当前值？

Question

有没有一种优雅的方法来更新地图中已经存在的值？

这看起来太可怕了：

val a = map.get ( something ) 
if ( a != null ) // case .. excuse my old language
   a.apply( updateFunction )
else 
   map.put ( something, default )

Answer 1

大多数时候，您可以插入可以在创建时更新的内容（例如，如果它是一个计数，则将 0 放入其中，然后更新为 1，而不是仅将 1 放入开始）。在这种情况下，

map.getOrElseUpdate(something, default).apply(updateFunction)

在极少数情况下，您无法以这种方式组织事物，

map(something) = map.get(something).map(updateFunction).getOrElse(default)

（但你必须参考something两次）。

Answer 2

这是我通常写的......不确定是否有更好的解决方案。

map.get(key) match {
  case None => map.put(key, defaultValue)
  case Some(v) => map(key) = updatedValue
}

事实上update，put对于可变映射也是一样的，但我通常update在现有条目和put新条目上使用，只是为了便于阅读。

另一件事是，如果您可以在不检查密钥是否存在的情况下弄清楚最终值是什么，您可以简单地编写map(key) = value，它会自动创建/替换条目。

最后，像这样的语句map(key) += 1实际上是有效的Map（这通常适用于具有update函数的集合），许多简单的数字运算也是如此。\

---

要解决双重放置，请使用可变对象而不是不可变值：

class ValueStore(var value: ValueType)
val map = new Map[KeyType, ValueStore]
...
map.get(key) match {
  case None => map.put(key, new ValueStore(defaultValue))
  case Some(v) => v.value = updatedValue
}

正如我在评论中提到的，HashMapis的底层结构HashTable实际上使用了这种可变包装类方法。HashMap是一个更好的总结类，但有时您只需要进行重复计算。

Answer 3

我很愚蠢，你是（非常）正确的：

map.get(key) match {
  case None => map.put(key, defaultValue)
  case Some(v) => v.apply(updateFunction) // changes state of value
}

谢谢