arrays - (F#) 查找二维数组中元素的索引

Question

我目前在查找二维数组中特定元素的索引时遇到一些问题。我知道标准数组可以使用 findIndex，如下所示。

let wantedElemented = 5
let index = Array.findIndex(fun x -> x = wantedElemented) array

我的问题是，如何将其应用于二维数组(矩形)。有没有比遍历整个数组比较每个元素更有效的方法？

for row in 0 .. max do
        for col in 0 .. max do
            if array.[row,col] = wantedElement then 
            let index = (row,col)
            index
            else (-1,-1)

如果我确实必须迭代整个数组，我将如何在不使用选项类型的情况下处理 else 条件

Answer 1

回应您的评论:是的，数组不能与头尾列表匹配。但他们确实有迹象! :-)

递归函数不必对数据结构进行递归。递归路径可以由不同的东西定义。例如:为什么我们不从零到最大值(或返回)递归数组索引？

let find2D needle (arr: int [,]) = 
    let rec go x y =
          if   y >= arr.GetLength 1 then None
          elif x >= arr.GetLength 0 then go 0 (y+1)
          elif arr.[x,y] = needle   then Some (x,y)
          else go (x+1) y
    go 0 0

上述解决方案将逐行扫描数组，直到找到needle，此时它将立即返回。

或者，您可以自己生成一个可选索引的序列，然后使用Seq.tryPick选择该序列中非None的第一个元素:

let find2D needle (arr: int [,]) = Seq.tryPick id <| seq {
    for i in 0..(arr.GetLength 0 - 1) do
        for j in 0..(arr.GetLength 1 - 1) do
            if arr.[i,j] = needle 
                then yield Some (i,j) 
                else yield None
}

由于序列的工作方式(它们是惰性的)，这只会迭代直到找到第一个 Some，然后停止。这稍微更简单(就可读性而言)，但性能也比上面的普通递归解决方案稍差，因为在这里我们会产生创建和维护序列的开销。