将 java 风格重构为更具功能性的 scala 风格

Question

我想将用户到他们朋友的地图转换为朋友到他们朋友的地图。 输入用户名：字符串 因此，通过将每个 List[User] 设为键，将 Map[Username, List[User]] 映射到 Map[Username, List[Username]]。

我下面的代码非常类似于 java，我想重构它以使其更具 idomatic scala 代码。

val map: Map[String, List[User]] = {
  var m  = collection.mutable.Map[String, collection.mutable.ListBuffer[User]]()
  for(user <- users) {
    for(friend <- user.Friends) {
      if (m.contains(friend)) {
        //m.get(friend.username) += user
      } else {
        m += (friend -> collection.mutable.ListBuffer[User]())
        //m.get(friend.username) += user
      }
    }
  }
  m.map{e =>
    (e._1 -> e._2.toList)}.toMap
}

我开始尝试这个：

users.map( user =>
  user.friends.flatMap(friend =>

)

作为旁注，我如何附加到 ListBuffer，由于某种原因这对我不起作用：

m.get(friend.username) += user

而新建一个列表缓冲区和追加将如何完成？

Answer 1

也许你会喜欢

    val map: Map[String, List[User]] = {
        val m = new collection.mutable.HashMap[String, collection.mutable.Set[User]]() with scala.collection.mutable.MultiMap[String, User]
        for {
            user <- users
            friend <- user.Friends
        }   m.addBinding(friend.username,user)

        m.mapValues(x ⇒ x.toList).toMap
    }

第二个问题：方法 m.get(friend.username) 返回 Option[ListBuffer[User]]。选项不包含 +=。

更新

一线解决方案

    val map: Map[String, Seq[User]] = {
        users.flatMap(user ⇒ user.Friends.map(_ → user)).groupBy(_._1.username).mapValues(_.map(_._2))
    }