swift - 通用二维数组上的 Swift 方法

Question

当数组是二维数组时，我想在 Array 类型上创建一个扩展方法，例如。[[Int]]，但类型应该是通用的。我正在尝试做这样的事情：

extension Array [where Element : ???] {

  public func transposed() -> ??? {
     // ...
  }
}

标签： swiftgenerics

Answer 1

这在 Swift 4 中有效。

extension Array where Element: Collection {

    public func transposed() -> [[Element.Iterator.Element]] {
        var result : [[Element.Iterator.Element]] = [[]]
        for row in self {
            for (y,column) in row.enumerated() {
                while (result.count <= y) {
                    result.append([])
                }
                result[y].append(column)
            }
        }
        return result
    }
}

然后单元测试也通过了：

func testTransposed() {
    XCTAssertEqual([[1]].transposed(), [[1]])
    XCTAssertEqual([[1,2,3],[4,5]].transposed(), [[1,4],[2,5],[3]])
    XCTAssertEqual([[4,1],[5,2],[3]].transposed(),[[4,5,3],[1,2]])
    XCTAssertEqual([[1,2,3]].transposed(), [[1],[2],[3]])
    XCTAssertEqual([[1],[2],[3]].transposed(),[[1,2,3]])
}