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

Swift:subscript 自定义下标

Swift:subscript 自定义下标 - 敏捷大拇指 - Swift:subscript 自定义下标





1、基本使用

[Swift] 纯文本查看 复制代码
struct TimesTable {  
    let multiplier: Int  
      
    subscript(index: Int) -> Int {  
        return multiplier * index  
    }  
}


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

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

[Swift] 纯文本查看 复制代码
let threeTimesTable = TimesTable(multiplier: 3)  
println("six times three is \(threeTimesTable[7])")





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

[Swift] 纯文本查看 复制代码
var numberOfLegs = ["spider":8,"ant":6,"cat":4]  
numberOfLegs["ant"] = nil  
println(numberOfLegs.count)


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




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

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

[Swift] 纯文本查看 复制代码
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中取值。

调用效果如下:

[Swift] 纯文本查看 复制代码
var matrix = Matrix(rows: 3, cols: 4)  
matrix[2, 1] = 3.4  
matrix[1, 2] = 5  
//  
var some2 = matrix[1, 2]  
println("some:\(some2)")  





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

[Swift] 纯文本查看 复制代码
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数组:

[Swift] 纯文本查看 复制代码
var arr = [1, 2, 3, 4, 5]  


input数组:

[Swift] 纯文本查看 复制代码
var input = [0,2]  


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

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

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




作者:秋恨雪