8

我有一个父键列表,每个父键都可能有零个或多个关联值。我不确定要使用哪个集合。

我在用Map[Int,List[String]]

我将地图声明为

var nodes = new HashMap[Int, List[String]]

然后我有两种方法来处理添加新元素。第一个是添加新键addNode,第二个是添加新值addValue。最初,该键将没有任何与之关联的值。稍后,在执行期间,将关联新值。

def addNode(key: Int) = nodes += (key -> "")

def addValue(key: Int, value: String) = ???

我不确定如何实施addValues

更新:

回应@oxbow-lakes 的回答,这是我收到的错误。请注意,键不需要有与之关联的值。

scala> var nodes = Map.empty[Int, List[String]]
nodes: scala.collection.immutable.Map[Int,List[String]] = Map()

scala> nodes += (1->null)

scala> nodes += (1 -> ("one" :: (nodes get 1 getOrElse Nil)))
java.lang.NullPointerException
    at .<init>(<console>:9)
    at .<clinit>(<console>)
    at .<init>(<console>:11)
    at .<clinit>(<console>)
    at $print(<console>)
    at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
    at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:39)
    at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:25)
    at java.lang.reflect.Method.invoke(Method.java:597)
    at scala.tools.nsc.interpreter.IMain$ReadEvalPrint.call(IMain.scala:704)
    at scala.tools.nsc.interpreter.IMain$Request$$anonfun$14.apply(IMain.scala:920)
    at scala.tools.nsc.interpreter.Line$$anonfun$1.apply$mcV$sp(Line.scala:43)
    at scala.tools.nsc.io.package$$anon$2.run(package.scala:25)
    at java.lang.Thread.run(Thread.java:680)

更新 2:

上面代码的问题是nodes += (1->null)键应该与之关联的行Nil。下面是工作代码。

scala> var nodes = Map.empty[Int, List[String]]
nodes: scala.collection.immutable.Map[Int,List[String]] = Map()

scala> nodes += (1->Nil)

scala> nodes += (1 -> ("one" :: (nodes get 1 getOrElse Nil)))

scala> nodes
res27: scala.collection.immutable.Map[Int,List[String]] = Map(1 -> List(one))
4

3 回答 3

25

使用多图

您可能想要使用MultiMap,它是一个与 同构的可变集合Map[K, Set[V]]。使用如下:

import collection.mutable
val mm = new mutable.HashMap[Int, mutable.Set[String]] with mutable.MultiMap[Int, String]

然后添加节点:

mm addBinding (key, value)

没有多重映射

另一种方法是坚持不可变的值。假设您想避免使用镜头(请参阅scalaz),您可以按如下方式添加节点:

nodes += (key -> (value :: (nodes get key getOrElse Nil)))

它在这里工作(响应您的评论):

scala> var nodes = Map.empty[Int, List[String]]
nodes: scala.collection.immutable.Map[Int,List[String]] = Map()

scala> def addNode(key: Int, value: String) =
     | nodes += (key -> (value :: (nodes get key getOrElse Nil)))
addNode: (key: Int, value: String)Unit

scala> addNode(1, "Hi")

scala> addNode(1, "Bye")

scala> nodes
res2: scala.collection.immutable.Map[Int,List[String]] = Map(1 -> List(Bye, Hi))

使用 Scalaz

使用 scalaz 库,您可以意识到这只是使用Empty模式:

nodes += (key -> (value :: ~(nodes get key)))

或者你可以利用Map一个monoid的事实:

nodes = nodes |+| Map(key -> List(value))
于 2012-05-13T12:45:35.293 回答
1

除了@oxbow_lakes 的回答之外,这里还有一个关于如何使用addMap将两个映射正确添加在一起的方法的想法(即,组合匹配键的列表,为新键添加新列表):

class EnhancedListMap(self: Map[Int,List[String]]) {
  def addMap(other: Map[Int,List[String]]) =
    (this.ungroup ++ enhanceListMap(other).ungroup)
      .groupBy(_._1)
      .mapValues(_.map(_._2))

  def ungroup() =
    self.toList.flatMap{ case (k,vs) => vs.map(k -> _) }
}

implicit def enhanceListMap(self: Map[Int,List[String]]) = new EnhancedListMap(self)

你会像这样使用它:

val a = Map(1 -> List("a","b"), 2 -> List("c","d"))
val b = Map(2 -> List("e","f"), 3 -> List("g","h"))
a addMap b
//Map(3 -> List(g, h), 1 -> List(a, b), 2 -> List(c, d, e, f))

您可以包括addNode, addValue, 和addValues相同的方式(到EnhancedListMap上面):

  def addNode(key: Int) =
    if(self contains key) self else self + (key -> Nil)

  def addValue(key: Int, value: String) =
    self + (key -> (value :: (self get key getOrElse Nil)))

  def addValues(key: Int, values: List[String]) =
    self + (key -> (values ::: (self get key getOrElse Nil)))

然后一起使用它们:

var nodes = Map.empty[Int, List[String]]             
// Map()
nodes = nodes.addNode(1)                             
// Map(1 -> List())
nodes = nodes.addValue(1,"a")                        
// Map(1 -> List(a))
nodes = nodes.addValue(2,"b")                        
// Map(1 -> List(a), 2 -> List(b))
nodes = nodes.addValues(2,List("c","d"))             
// Map(1 -> List(a), 2 -> List(c, d, b))
nodes = nodes.addValues(3,List("e","f"))             
// Map(1 -> List(a), 2 -> List(c, d, b), 3 -> List(e, f))
nodes = nodes.addMap(Map(3 -> List("g","h"), 4-> List("i","j")))
// Map(1 -> List(a), 2 -> List(c, d, b), 3 -> List(e, f, g, h), 4 -> List(i, j))
于 2012-05-13T14:16:22.497 回答
0

我很喜欢getOrElseUpdate可变映射提供的方法:

import scala.collection.mutable._

private val nodes = new HashMap[Int, Buffer[String]]

def addNode(key: Int): Unit =
  nodes.getOrElseUpdate(key, new ArrayBuffer)

def addValue(key: Int, value: String): Unit  =
  nodes.getOrElseUpdate(key, new ArrayBuffer) += value
于 2012-05-13T22:36:04.873 回答