Swift4.0 二维数组

3,131 阅读2分钟
原文链接: www.jianshu.com

在 C语言或者 OC 中生成9行7列的cookies你可以这样写

int cookies[9][7]

这生成了具有63个元素的二维数组,如果要找到第三行第六列中的元素可以这样写

 myCookie = cookies[3][6];

但在 swift 中 这种方式是不可以的.在 swift 中创建多维的数组,可以这样写

var cookies = [[Int]]()
for _ in 1...9 {
  var row = [Int]()
  for _ in 1...7 {
    row.append(0)
  }
  cookies.append(row)
}

可以用

let myCookie = cookies[3][6]

找到3行6列的元素.

也可以用一行代码生成
var cookies = [[Int]](repeating: [Int](repeating: 0, count: 7), count: 9)
这样看起来比较复杂,你也可以用一个函数来创建
func dim<T>(_ count: Int, _ value: T) -> [T] {
  return [T](repeating: value, count: count)
}

然后可以这样写生成一个二维数组

var cookies = dim(9, dim(7, 0))

Swift推断数组的数据类型必须是Int,因为您指定了0作为数组元素的默认值。要使用字符串,你可以写:

var threeDimensions = dim(2, dim(3, dim(4, 0)))

但是 以这种方式使用多维数组或多个嵌套数组的缺点是,无法了解维度代表什么。 相反,用下面的方式您可以更方便创建就像一个2d数组一样的自己定义的数组类型 :

public struct Array2D<T> {
  public let columns: Int
  public let rows: Int
  fileprivate var array: [T]
  
  public init(columns: Int, rows: Int, initialValue: T) {
    self.columns = columns
    self.rows = rows
    array = .init(repeating: initialValue, count: rows*columns)
  }
  
  public subscript(column: Int, row: Int) -> T {
    get {
      precondition(column < columns, "Column \(column) Index is out of range. Array<T>(columns: \(columns), rows:\(rows))")
      precondition(row < rows, "Row \(row) Index is out of range. Array<T>(columns: \(columns), rows:\(rows))")
      return array[row*columns + column]
    }
    set {
      precondition(column < columns, "Column \(column) Index is out of range. Array<T>(columns: \(columns), rows:\(rows))")
      precondition(row < rows, "Row \(row) Index is out of range. Array<T>(columns: \(columns), rows:\(rows))")
      array[row*columns + column] = newValue
    }
  }
}

Array2D 结构体是一个通用的类型,它可以创建任何类型的的数组元素不仅仅是数子.

// 创建一个二维数组的实例
var cookies = Array2D(columns: 9, rows: 7, initialValue: 0)

运用下标==subscript==函数可以检索数组中的值

let myCookie = cookies[column, row]

也可以改变数组中的某一个值

cookies[column, row] = newCookie

在内部,Array2D使用单个一维数组来存储数据。该数组中的对象的索引是由((行x numberofcolumn)+列 )给出的,但是作为Array2D的使用者,您只需要考虑“列”和“行”,细节将由Array2D来完成。这是将原始类型包装成包装器类或结构体的优点