Wie findet man den Index eines Artikels in einem multidimensionalen Array schnell?

Question

Lassen Sie uns sagen, dass ich dieses Array haben:

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

Jetzt möchte ich etwas wie folgt aus:

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

so, dass ich es so nennen kann:

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

Natürlich, ich kann verwenden:

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

Aber das ist nicht mal knapp!

Ich möchte einen Weg, dies zu tun, die swifty ist. Ich denke, vielleicht kann ich reduce oder map verwenden?

Answer 1

Sie können Ihren Code leicht mit enumerate() und indexOf() vereinfachen. Auch die Funktion sollte ein optionales Tupel zurückgeben, da das Element möglicherweise nicht in der "Matrix" vorhanden ist. Schließlich können Sie es Generika machen:

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) 
} 

Swift 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 
    } 
}