如何快速查找多維數組中項目的索引？

Question

比方說，我有這樣的數組：

let a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 

現在，我想是這樣的：

public func indicesOf(x: Int, array: [[Int]]) -> (Int, Int) { 
     ... 
    } 

，這樣我可以這樣調用：

indicesOf(7, array: a) // returns (2, 0) 

當然，我可以使用：

for i in 0..<array.count { 
    for j in 0..<array[i].count { 
     if array[i][j] == x { 
      return (i, j) 
     } 
    } 
} 

但這甚至不太接近swifty！

我想要一個方法來做到這一點很快速。我想也許我可以使用reduce或map？

Answer 1

您可以用enumerate()和indexOf()稍微簡化您的代碼。 此外，函數應該返回一個可選的元組，因爲元素 可能不存在於「矩陣」中。最後，你可以把它通用：

func indicesOf<T: Equatable>(x: T, array: [[T]]) -> (Int, Int)? { 
    for (i, row) in array.enumerate() { 
     if let j = row.indexOf(x) { 
      return (i, j) 
     } 
    } 
    return nil 
} 

你也可以把它嵌套的ArrayEquatable 元素的擴展：

extension Array where Element : CollectionType, 
    Element.Generator.Element : Equatable, Element.Index == Int { 
    func indicesOf(x: Element.Generator.Element) -> (Int, Int)? { 
     for (i, row) in self.enumerate() { 
      if let j = row.indexOf(x) { 
       return (i, j) 
      } 
     } 
     return nil 
    } 
} 

if let (i, j) = a.indicesOf(7) { 
    print(i, j) 
} 

斯威夫特3：

extension Array where Element : Collection, 
    Element.Iterator.Element : Equatable, Element.Index == Int { 

    func indices(of x: Element.Iterator.Element) -> (Int, Int)? { 
     for (i, row) in self.enumerated() { 
      if let j = row.index(of: x) { 
       return (i, j) 
      } 
     } 
     return nil 
    } 
}