在Scala中反转/转置一对多地图

Question

将a Map[A, Set[B]]变为a 的最佳方法是Map[B, Set[A]]什么？

例如,我该怎么转

Map(1 -> Set("a", "b"),
    2 -> Set("b", "c"),
    3 -> Set("c", "d"))

进入

Map("a" -> Set(1),
    "b" -> Set(1, 2),
    "c" -> Set(2, 3),
    "d" -> Set(3))

(我在这里只使用不可变集合.而我真正的问题与字符串或整数无关.:)

Answer 1

在aioobe和Moritz的帮助下:

def reverse[A, B](m: Map[A, Set[B]]) =
  m.values.toSet.flatten.map(v => (v, m.keys.filter(m(_)(v)))).toMap

如果明确调用contains,它会更具可读性:

def reverse[A, B](m: Map[A, Set[B]]) =
  m.values.toSet.flatten.map(v => (v, m.keys.filter(m(_).contains(v)))).toMap