通常情况下。我们在使用数组(Array)或字典(Dictionary)时会使用到下标。事实上在Swift中。我们还能够给类、结构、枚举等自己定义下标(subscript)。
一、基本使用
struct TimesTable {
let multiplier: Int
subscript(index: Int) -> Int {
return multiplier * index
}
}
我在TimesTable这个结构中自己定义了一个subscript。而且这个subscript类似于方法,看上去它的类型为 Int -> Int。
然后在调用的时候。就能够使用"[index]"这种形式取值。
let threeTimesTable = TimesTable(multiplier: 3)
println("six times three is \(threeTimesTable[7])")
二、使用subscript能够删除字典中的某个特定的key-value
var numberOfLegs = ["spider":8,"ant":6,"cat":4]
numberOfLegs["ant"] = nil
println(numberOfLegs.count)
上面的numberOfLegs初始时候它有三对值,当进行numberOfLegs["ant"] = nil 操作后,相当于key为"ant"的key-value被删除了。所以打印的结果为2。
三、subscript中的索引參数不一定永远是一个Int类型的index,它也能够有多个參数。
比如。我们能够使用subscript将一维数组模拟成二维数组。
struct Matrix {
let rows: Int
let cols: Int
var grid: [Double]
init(rows: Int, cols: Int) {
self.rows = rows
self.cols = cols
self.grid = Array(count: rows * cols, repeatedValue: 0.0)
}
func indexIsValidForRow(row: Int, col: Int) -> Bool {
return row >= 0 && row < rows && col >= 0 && col < cols;
}
subscript(row: Int, col: Int) -> Double {
get {
assert(indexIsValidForRow(row, col: col), "index out of range")
return grid[row * cols + col]
}
set {
assert(indexIsValidForRow(row, col: col), "index out of range")
grid[row * cols + col] = newValue
}
}
}
代码中的grid成员属性是一个含有rows * cols 个元素的一维数组。
然后定义一个subscript, 这里的下标有两个:row和col。然后依据详细的输入參数,从grid数组中取出相应的值。
所以这里的下标仅仅是模拟。看起来输入row和col,但实际还是从一维数组grid中取值。
调用效果例如以下:
var matrix = Matrix(rows: 3, cols: 4)
matrix[2, 1] = 3.4
matrix[1, 2] = 5
//
var some2 = matrix[1, 2]
println("some:\(some2)")
四、获取数组中指定索引位置的子数组,我们能够在Array的扩展中用subscript来实现。
extension Array {
subscript(input: [Int]) -> ArraySlice<T> {
get {
var array = ArraySlice<T>()
for i in input {
assert(i < self.count, "index out of range")
array.append(self[i])
}
return array
}
set {
// i表示数组input自己的索引,index表示数组self的索引
for (i, index) in enumerate(input) {
assert(index < self.count, "index out of range")
self[index] = newValue[i]
}
}
}
}
代码中的input数组表示选取的Array中的某些下标。比如:
arr数组
var arr = [1, 2, 3, 4, 5]
input数组
var input = [0,2]
那么通过input中的值在arr数组中取得的子数组为 [1, 3]
subscript中的get方法就是依据input中的数组的下标值取得arr数组中相应下标的子数组。
subscript中的set方法就是依据input中的数组的下标值对arr数组中相应下标的内容又一次赋值。