scala - 在 Scala 中以函数方式迭代表填充 2 个 HasSet

Question

我正在编写一个 Scala 应用程序，并试图以一种功能方式遍历一个表(以二维数组的形式)。对于表中的每一行，我想用第一列中的所有不同值填充一个集合，并用第二列中的所有不同值填充第二个集合。

我尝试了很多方法，但找不到任何解决方案如何以功能风格来做到这一点。因为我从迭代中获得了 2 个新变量，如果没有非功能性帮助，这似乎是不可能的。

这是一个非功能性示例，其中包含用于包含产品和客户的表的可变 HashSet:

val myInputTable =
  Array(Array("Product A","Customer 1"), Array("Product B","Customer 1"),
    Array("Product C","Customer 2"), Array("Product A","Customer 2"))

val productSet = new collection.mutable.HashSet[String]
val customerSet = new collection.mutable.HashSet[String]

for(
  inputLine <- myInputTable;
  inputElement <- inputLine
) {
  if (inputLine.indexOf(inputElement) == 0) {
    productSet.add(inputElement)
  } else {
    customerSet.add(inputElement)
  }
}

println("Product Set:")
productSet.foreach(println)
println("\nCustomer Set:")
customerSet.foreach(println) 

有没有办法用不可变的集合、其他对象或者 for-yield 语法来做到这一点？

感谢您的任何答案或提示，

菲利克斯

Answer 1

任何时候你发现自己试图对一些迭代序列的代码进行 FP-ify，同时更新一些可变状态，一个好的第一种方法是使用 foldLeft :

val myInputTable =
  Array(Array("Product A","Customer 1"), Array("Product B","Customer 1"),
    Array("Product C","Customer 2"), Array("Product A","Customer 2"))

val (products, customers) =
  myInputTable.foldLeft((Set.empty[String], Set.empty[String])) {
    case ((ps, cs), Array(p, c)) => (ps + p, cs + c)
    case ((ps, cs), _) => (ps, cs) // Alternatively fail here.
  }

foldLeft 的第一个参数是初始状态。我们想要使用两个不可变的集合，所以我们使用了 Set.empty[String] 的元组。 . foldLeft 的下一个参数这是一个函数:

  {
    case ((ps, cs), Array(p, c)) => (ps + p, cs + c)
    case ((ps, cs), _) => (ps, cs) // Alternatively fail here.
  }

这应该是当前累积状态的函数 (ps, cs)和每个元素 Array(p, c)到下一个状态。它将从左到右应用于集合中的每个函数(因此是 foldLeft )，累积状态变化，并返回状态的最终值。它是这样工作的:

scala> val (products, customers) =
     |   myInputTable.foldLeft((Set.empty[String], Set.empty[String])) {
     |     case ((ps, cs), Array(p, c)) => (ps + p, cs + c)
     |     case ((ps, cs), _) => (ps, cs) // Alternatively fail here.
     |   }
products: scala.collection.immutable.Set[String] = Set(Product A, Product B, Product C)
customers: scala.collection.immutable.Set[String] = Set(Customer 1, Customer 2)

在某些情况下，可能会有更具体的组合子让您更简洁地表达您的操作，但是 foldLeft是一个很好的通用起点，它允许从可变代码到纯函数代码的相当简单的转换。