Swift：subscript

通常情况下，我们在使用数组（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,"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数组中对应下标的内容重新赋值。