Scala - reduce / foldLeft

时间:2011-10-11 08:06:09

标签: scala map reduce fold

我有一个嵌套的地图m,就像:

m = Map("email" -> "a@b.com", "background" -> Map("language" -> "english"))

我有一个数组arr = Array("background","language")

如何foldLeft / reduce数组并从地图中找到字符串“english”。我试过这个:

arr.foldLeft(m) { (acc,x) => acc.get(x) }

但是我收到了这个错误:

<console>:10: error: type mismatch;
 found   : Option[java.lang.Object]
 required: scala.collection.immutable.Map[java.lang.String,java.lang.Object]
       arr.foldLeft(m) { (acc,x) => acc.get(x) }

2 个答案:

答案 0 :(得分:12)

你应该注意类型。在这里,您从m : Map[String, Any]开始作为您的acc。您与字符串x合并并调用get,后者返回Option[Object]。要继续,您必须检查是否有值,检查此值是否为Map,强制转换(由于类型擦除而未选中,因此很危险)。

我认为错误在于你的结构类型Map [String,Any]代表你所拥有的东西相当糟糕。

假设您改为

sealed trait Tree
case class Node(items: Map[String, Tree]) extends Tree
case class Leaf(s: String) extends Tree

您可以添加一些助手来宣布树容易

object Tree {
  implicit def fromString(s: String) = Leaf(s)
  implicit def fromNamedString(nameAndValue: (String, String)) 
    = (nameAndValue._1, Leaf(nameAndValue._2))
}
object Node {
  def apply(items: (String, Tree)*) : Node = Node(Map(items: _*))
}

然后声明树就像第一个版本一样简单,但类型更精确

m = Node("email" -> "a@b.com", "background" -> Node("language" -> "english"))

然后,您可以添加方法,例如在trait Tree

  def get(path: String*) : Option[Tree] = {
    if (path.isEmpty) Some(this)
    else this match {
      case Leaf(_) => None
      case Node(map) => map.get(path.head).flatMap(_.get(path.tail: _*))
    }
  }
  def getLeaf(path: String*): Option[String] 
    = get(path: _*).collect{case Leaf(s) =>s}

或者你宁愿用折叠

来做
  def get(path: String*) = path.foldLeft[Option[Tree]](Some(this)) {
    case (Some(Node(map)), p) => map.get(p)
    case _ => None
  }

答案 1 :(得分:4)

实际上不支持折叠作为嵌套地图的抽象。此外,您正在以一种阻止类型系统为您提供帮助的方式接近它。但是,如果你坚持,那么你想要一个递归函数:

def lookup(m: Map[String,Object], a: Array[String]): Option[String] = {
  if (a.length == 0) None
  else m.get(a(0)).flatMap(_ match {
    case mm: Map[_,_] => lookup(mm.asInstanceOf[Map[String,Object]],a.tail)
    case s: String if (a.length==1) => Some(s)
    case _ => None
  })
}