如何更新mutable.Map中的当前值?

时间:2013-01-25 19:06:36

标签: scala

是否有一种优雅的方式来更新Map中已存在的值?

这看起来太吓人了:

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

3 个答案:

答案 0 :(得分:11)

大多数情况下,您可以插入可以在创建时更新的内容(例如,如果它是计数,则将0放入其中,然后更新为1而不是仅将1放入开始)。在那种情况下,

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

在极少数情况下,你无法以这种方式组织事情,

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

(但你必须两次引用something

答案 1 :(得分:3)

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

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

事实上,updateput对于可变地图是相同的,但我通常在现有条目上使用update而在新条目中使用put,只是为了便于阅读。< / p>

另一件事是,如果你能在不检查密钥存在的情况下弄清楚最终价值是什么,你可以简单地写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
}

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

答案 2 :(得分:0)

我是愚蠢的,你(非常)对:

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

TSTS 感谢