折叠基于公共元素的集合列表

Question

我有一些看起来很简单的问题，我想使用 Scala 功能性地解决它，但无法解决。 我正在学习 Scala，并希望确保它使用函数式编程原则完成。下面的一段代码（甚至不起作用......）是我管理的所有内容 它仍然有一个可变变量... 任何帮助表示赞赏。

基本上我有一个集合列表，我想加入所有具有共同元素的集合。 并递归应用，即继续加入列表中的集合，直到它们只剩下不相交的集合。

所以对于输入 List(Set(1,2,3), Set(4,1,5), Set(6,2,7), Set(8,9))我想得到List(Set(1,2,3,4,5,6,7), Set(8,9))。

下面是我的代码 - 再次无法完全正常工作。

    def mergeSets(sets: List[Set[Int]]): List[Set[Int]] = {
        @tailrec
        def go(list: List[Set[Int]], result: List[Set[Int]]): List[Set[Int]] = list match {
          case Nil => result
          case h :: t => {
            var tmpResult = result
            tmpResult = list.filter(_.intersect(h).nonEmpty).map(_ ++ h) ::: result
            tmpResult = list.filter(_.intersect(h).isEmpty) ::: tmpResult
            go(t, tmpResult.toSet.toList)
          }
        }
        go(sets, List())
      }

Answer 1

这是结合高阶函数以获得所需结果的“单线”解决方案：

val l = val l = List(Set(1,2,3), Set(4,1,5), Set(6,2,7), Set(8,9))
(for (el <- l) yield 
     l.foldLeft(el)((s, x) =>
            if (s.intersect(x).size > 0) s.union(x) else s)
).toSet

主要思想是遍历每个集合并产生所有适当集合的并集。这里唯一的技巧是使用toSet 来删除重复项。

Answer 2

我会使用元组返回类型而不是 List，因为您希望该位置有意义。我在下面使用的策略是折叠每个集合，对于每个集合，如果有另一个集合具有任何共同元素，则添加到第一个元组位置，否则添加到第二个。

def mergeSets(sets: Set[Int]*): (Set[Int], Set[Int]) = {
  sets.foldLeft((Set.empty[Int], Set.empty[Int])) { case ((c, d), n) =>
    if (sets.exists(s => n.exists(s.contains) && s != n))
      (c ++ n, d)
    else
      (c, d ++ n)
  }
}

val (common, disjointed) = mergeSets(Set(1,2,3), Set(4,1,5), Set(6,2,7), Set(8,9))

println(common)
// Set(5, 1, 6, 2, 7, 3, 4)


println(disjointed)
// Set(8, 9)