Scala中的函数式编程：在多维列表中查找元素的索引

Question

我有一个二维列表ll : List[List[T]] 和一个函数f : T => Boolean。我想找到一个整数元组(i, j)，这样f(l(i)(j)) == true 只要它存在，就可能包裹在Option 中，并且想以函数式编程的方式来做这件事。我目前的工作解决方案如下：

ll.zipWithIndex
  .flatMap{case (l, i) => l.map((_, i)).zipWithIndex}
  .find{case ((l, _), j) => f(l(j))}
  .map{case ((_, i), j) => (i, j)}

但是，我觉得应该有更好的方法来实现这一点 - 我的解决方案非常麻烦，并且不能很好地推广到高维数组。那么有没有更好的方法来做到这一点？

在我走之前：我完全知道这很容易通过嵌套的 for 循环实现。所以我只对函数式编程风格的解决方案感兴趣。

Answer 1

(for { 
  (l, i) <- ll.zipWithIndex
  (e, j) <- l.zipWithIndex
  if f(e)
} yield (i, j)).headOption

Answer 2

您没有提供任何用于测试的示例输入，但这似乎可行。它返回一个IndexedSeq[(Int, Int)] 集合，应该（我认为）包含所有(i,j) 对，其中f(ll(i)(j)) 返回true。

ll.indices
  .flatMap(i => ll(i).indices
                     .collect{case j if f(ll(i)(j)) => (i,j)})

和这个差不多……

for {
  i <- ll.indices
  j <- ll(i).indices
  if f(ll(i)(j))
} yield (i,j)

...并且，如 cmets 中所述，完全兼容 FP。