Swift：subscript

本文转载自：http://blog.csdn.net/sinat_27706697/article/details/47122137 感谢作者：秋恨雪

通常情况下，我们在使用数组（Array）或字典（Dictionary）时会使用到下标。其实在Swift中，我们还可以给类、结构、枚举等自定义下标（subscript）。

一、基本使用

1. struct TimesTable {
2. let multiplier: Int
3. subscript(index: Int) -> Int {
4. return multiplier * index
5. }
6. }

我在TimesTable这个结构中自定义了一个subscript，并且这个subscript类似于方法，看上去它的类型为 Int -> Int。

然后在调用的时候，就可以使用"[index]"这样的形式取值。

1. let threeTimesTable = TimesTable(multiplier: 3)
2. println("six times three is \(threeTimesTable[7])")

二、使用subscript可以删除字典中的某个特定的key-value

1. var numberOfLegs = ["spider":8,"ant":6,"cat":4]
2. numberOfLegs["ant"] = nil
3. println(numberOfLegs.count)

上面的numberOfLegs初始时候它有三对值，当进行numberOfLegs["ant"] = nil 操作后，相当于key为"ant"的key-value被删除了，所以打印的结果为2。

三、subscript中的索引参数不一定永远是一个Int类型的index，它也可以有多个参数。

例如，我们可以使用subscript将一维数组模拟成二维数组。

1. struct Matrix {
2. let rows: Int
3. let cols: Int
4. var grid: [Double]
5. init(rows: Int, cols: Int) {
6. self.rows = rows
7. self.cols = cols
8. self.grid = Array(count: rows * cols, repeatedValue: 0.0)
9. }
10. func indexIsValidForRow(row: Int, col: Int) -> Bool {
11. return row >= 0 && row < rows && col >= 0 && col < cols;
12. }
13. subscript(row: Int, col: Int) -> Double {
14. get {
15. assert(indexIsValidForRow(row, col: col), "index out of range")
16. return grid[row * cols + col]
17. }
18. set {
19. assert(indexIsValidForRow(row, col: col), "index out of range")
20. grid[row * cols + col] = newValue
21. }
22. }
23. }

代码中的grid成员属性是一个含有rows * cols 个元素的一维数组。

然后定义一个subscript, 这里的下标有两个：row和col。然后根据具体的输入参数，从grid数组中取出对应的值。所以这里的下标只是模拟，看起来输入row和col，但实际还是从一维数组grid中取值。

调用效果如下：

1. var matrix = Matrix(rows: 3, cols: 4)
2. matrix[2, 1] = 3.4
3. matrix[1, 2] = 5
4. //
5. var some2 = matrix[1, 2]
6. println("some:\(some2)")

四、获取数组中指定索引位置的子数组，我们可以在Array的扩展中用subscript来实现。

1. extension Array {
2. subscript(input: [Int]) -> ArraySlice<T> {
3. get {
4. var array = ArraySlice<T>()
5. for i in input {
6. assert(i < self.count, "index out of range")
7. array.append(self[i])
8. }
9. return array
10. }
11. set {
12. // i表示数组input自己的索引，index表示数组self的索引
13. for (i, index) in enumerate(input) {
14. assert(index < self.count, "index out of range")
15. self[index] = newValue[i]
16. }
17. }
18. }
19. }

代码中的input数组表示选取的Array中的某些下标。例如：

arr数组

1. var arr = [1, 2, 3, 4, 5]

input数组

1. var input = [0,2]

那么通过input中的值在arr数组中取得的子数组为 [1, 3]

subscript中的get方法就是根据input中的数组的下标值取得arr数组中对应下标的子数组。

subscript中的set方法就是根据input中的数组的下标值对arr数组中对应下标的内容重新赋值。