Scala：比较一个巨大列表中的所有元素

Question

请就算法和实现提出建议，以比较 Scala 中很长列表中的元素。我有一个包含数千个字符串的列表（来自 SQL），我需要将每个列表元素与该列表中的所有其他元素进行比较。

因此，我需要获取元组列表：List[(String, String, Boolean)] 其中前两个元素是要匹配的字符串，第三个是结果。

对于 N 个元素的列表，到目前为止，我的算法如下：

1. 占据榜首
2. 将 head 与列表中剩余的 N-1 个元素进行比较
3. 从旧列表的尾部创建新列表，并使用这个包含 N -1 个元素的新列表完成上述所有工作：

代码：

   /**
   * Compare head of the list with each remaining element in this list
   */
  def cmpel(
    fst: String, lst: List[String],
    result: List[(String, String, Boolean)]): List[(String, String, Boolean)] = {

    lst match {
      case next :: tail => cmpel(fst, tail, (fst, next, fst == next) :: result)
      case nill => result.reverse
    }
  }

  /**
   * Compare list elements in all combinations of two
   */
  def cmpAll(lst: List[String],
    result: List[(String, String, Boolean)]): List[(String, String, Boolean)] = {
    lst match {
      case head :: tail => cmpAll(tail, result ++ cmpel(head, tail, List()))
      case nill => result
    }
  }

  def main(args: Array[String]): Unit = {
    val lst = List[String]("a", "b", "b", "a")
    println(cmpAll(lst, List()))
  }

结果：

 List((a,b,false), (a,b,false), (a,a,true), (b,b,true), (b,a,false), (b,a,false))

谢谢！

Answer 1

您可以使用tails 和flatMap 方法编写更简洁和惯用的解决方案：

list.tails.flatMap {
  case x :: rest => rest.map { y =>
    (x, y, x == y)
  }
  case _ => List()
}.toList

tails 方法返回一个迭代器，该迭代器将.tail 的重复应用程序迭代到列表中。迭代器中的第一个元素是列表本身，然后是列表的尾部，以此类推，最后返回空列表。