将列表的元素与所有其他元素进行比较

Question

假设我有一个 Scala 列表，

val list : List[Set[Int]] = List(set1, set2, set3, set4)

我列表中的每个元素都是一组整数。

我正在寻找一种优雅的方法来创建一个列表，其中我的列表的每个元素都与所有其他元素进行比较。例如，我可能想要生成一个列表，其中每个元素（集合）仅包含不包含在列表的任何其他元素中的元素。

例如我有一些功能：

def f(element : Set[Int], rest_of_list : List[Set[Int]]) : Set[Int]

我想做类似于 list.map(f(_, rest_of_list)) 的事情，而不必为每个元素构造 rest_of_list。我现在能想到的唯一方法是使用 slice，这使得它非常丑陋且效率低下，imo。

Answer 1

您可以使用 diff 代替 slice:

 list.map(set => set diff (list diff List(set)).flatten.toSet)
 //Alternative with filterNot
 list.map(set => set.filterNot((list diff List(set)).flatten.contains))

diff 在这里工作，因为它只删除元素的一个实例，并且 Set[Int] 有一个很好的 equals 方法：

List(Set(1), Set(1, 2)) diff List(Set(1)) //List(Set(1, 2))
List(Set(1), Set(1)) diff List(Set(1)) //List(Set(1))

因此，将一个集合映射到列表中所有其他集合的列表：

val setWithOthers = list.map(set => set -> (list diff List(set)))
val setWithOthers = (list, list.map(list diff List(_))).zipped.toList //equivalent, less efficient

因此，如果您想将函数 f 映射到 list 的元素，您可以这样做：

setsWithOthers.map { case(set, others) => f(set, others) }
setsWithOthers.map(kvp => f(kvp._1, kvp._2))
setsWithOthers.map((f _).tupled)

您还可以创建完整的多重集来跟踪每个元素出现的次数。然后，在循环的每次迭代中，您可以 "subtract" 来自完整多重集的一个集合：

val multiset = list.flatten.groupBy(identity).mapValues(_.length)

Answer 2

我不知道它是否像您想要的那样优雅，但可以用更通用的方式将 'f' 替换为任何东西：

val list = List(Set(1,2,3,4),Set(1,2),Set(1,3))

list.map(e => f(e,list.filterNot(_==e)))
//res0: List[Set[Int]] = List(Set(4), Set(), Set())

def f(element : Set[Int], rest_of_list : List[Set[Int]]) : Set[Int] = {
  (element.toList diff rest_of_list.flatten.toList).toSet
}

将列表的元素与所有其他元素进行比较

Compare an element of a list against all other elements

list

scala

scala-collections