使用 FoldLeft 在 Scala 中制作邻接表

Posted 2023-03-23

【中文标题】使用 FoldLeft 在 Scala 中制作邻接表

【英文标题】：Making Adjacency List in Scala with FoldLeft

【发布时间】：2021-08-17 18:12:10

【问题描述】：

给定这个函数

  def makeAdjacencyList(edgeList: Seq[(String, String)]): Map[String, Seq[String]] = 
    val emptyMap = Map[String, Seq[String]]().withDefaultValue(Seq[String]())

    edgeList.foldLeft(emptyMap) 
      case (acc, curr) => 
        acc + (curr._1 -> (acc(curr._1) :+ curr._2))
        acc + (curr._2 -> (acc(curr._2) :+ curr._1))
      
    
  

它接受一个无向图的edgeList，例如：

Vector((S1,E1), (S1,D1), (S2,E2), (S2,D1), (S3,E1), (S3,D2))

然而该函数给出的输出为

[E1 -> [S1, S3], D1 -> [S1, S2], E2 -> [S2], D2 -> [S3]]

但我期待的输出是

[S1 -> [E1, D1], S2 -> [E2, D1], S3 -> [E1], E1 -> [S1, S3], D1 -> [S1, S2], E2 -> [S2], D2 -> [S3]]

foldLeft (acc + (curr._1 -> (acc(curr._1) :+ curr._2))) 的第一行似乎被忽略了，我不太清楚为什么。

【参考方案1】：

case 块中的两行代码都会产生一个结果，但只返回第二个结果并成为下一个 acc 值。

这会解决它。

edgeList.foldLeft(emptyMap) 
  case (acc, curr) =>
    acc + (curr._1 -> (acc(curr._1) :+ curr._2)) +
          (curr._2 -> (acc(curr._2) :+ curr._1))

另一方面，这是实现所需输出的不同方法。

def makeAdjacencyList(edgeList: Seq[(String,String)]): Map[String,Seq[String]] =
  edgeList.flatMap(edge => Seq(edge, edge.swap))
          .groupMap(_._1)(_._2)