概述

我正在尝试将以下函数转换为2D数组的通用扩展. func rotate(_ input: [[Int]]) -> [[Int]] { let length = input[0].count var value = Array(repeaTing: [Int](), count: length) for index in 0 ..< length { value[index]

我正在尝试将以下函数转换为2D数组的通用扩展.

func rotate(_ input: [[Int]]) -> [[Int]]
{
  let length = input[0].count
  var value = Array(repeaTing: [Int](),count: length)
  for index in 0 ..< length
  {
    value[index] = input.map { $0[index] }.reversed()
  }
  return value
}

我特别难以指出如何指定约束以允许我访问第二个维度.这是一次失败的尝试：

extension Array where Element: Collection,Element.Iterator.Element: Collection
{
  private func rotate()
  {
    let count = self[0].count // Element.IndexDistance instead of Int

    // Expression type 'Array<Element>' is ambiguous without more context
    var returnValue = Array(repeaTing: Element,count: 11)
    for index in 0 ..< count // Element.IndexDistance instead of Int
    {
      returnValue[index] = self.map { $0[index] }.reversed()
    }
    return returnValue
  }
}

问题是编译器不知道你的扩展是用于2D数组 – 它只知道它是用于集合数组.因此,关联类型IndexDistance和Index不一定是Int.

因此,解决方案是约束您的扩展,以便Element的IndexDistance和Index属于Int类型.这将允许您形成范围0 ..< count,因为count现在将是Int(IndexDistancE)类型 - 并且您将能够使用Ints下标地图(_ :)中的元素(因为下标需要指数). (一旦支持 concrete same-type requirements,这将是微不足道的,因为您可以简单地将Element约束为数组,但这还不可能.)

您还应该注意,您的约束Element.Iterator.Element：Collection是不正确的,因为它会将扩展约束到集合的3D数组(元素是集合的数组,其中该集合的元素是集合).

最后,您可能必须定义一个类型来表示2D数组的“内部元素”类型,因为Swift当前具有some limitations when working with nested types directly,例如在创建该类型的空2D阵列时.

因此,当前方法的工作版本如下所示：

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

    private func rotate() -> [[Element.Iterator.Element]] {

        typealias InnerElement = Element.Iterator.Element

        // in the case of an empty array,simply return an empty array
        if self.isEmpty { return [] } 
        let length = self[0].count

        var returnValue = [[InnerElement]](repeaTing: [InnerElement](),count: length)
        for index in 0..<length {
            returnValue[index] = self.map{ $0[index] }.reversed()
        }
        return returnValue
    }
}

通过使用嵌套映射(_ :)可以大大简化@MartinR points out below,从而消除了对类型的需求,因为我们不再需要创建“结果”数组：

private func rotate() -> [[Element.Iterator.Element]] {

    if self.isEmpty { return [] }
    let length = self[0].count

    return (0..<length).map { index in
        self.map { $0[index] }.reversed()
    }
}

虽然请注意,扩展的约束仅适用于Int索引并不是绝对必要的(但是没有实际的区别,因为您只打算将其用于2D数组).另一种方法是直接迭代内部集合的索引.

为了做到这一点,你只需要约束你的扩展,以便内部集合的指数具有与集合的索引相同类型的元素(在数组的情况下,指数是一个CountableRange< Int>)：

extension Array where Element: Collection,Element.InDices.Iterator.Element == Element.Index {
    private func rotate() -> [[Element.Iterator.Element]] {

        if self.isEmpty { return [] }

        return self[0].inDices.map { index in
            self.map { $0[index] }.reversed()
        }
    }
}