Swift 中如何定义下标？

推荐答案

在 Swift 中，下标（Subscript）可以通过在类、结构体或枚举中定义 subscript 关键字来实现。下标允许你通过索引或键来访问集合、列表或序列中的元素。下标的定义语法如下：

subscript(index: Int) -> Element {
    get {
        // 返回与索引对应的值
    }
    set(newValue) {
        // 设置与索引对应的值
    }
}

其中：

• index 是下标的参数，可以是任意类型。
• Element 是返回值的类型。
• get 块用于获取值，set 块用于设置值（可选）。

本题详细解读

1. 下标的基本用法

下标可以用于类、结构体或枚举中，允许你通过索引或键来访问或修改集合中的元素。例如，你可以为自定义类型定义一个下标来访问数组中的元素：

// www.javascriptcn.com code example
struct MyArray {
    var elements: [Int]
    
    subscript(index: Int) -> Int {
        get {
            return elements[index]
        }
        set(newValue) {
            elements[index] = newValue
        }
    }
}

var myArray = MyArray(elements: [1, 2, 3, 4])
print(myArray[2]) // 输出: 3
myArray[2] = 10
print(myArray[2]) // 输出: 10

2. 只读下标

如果你只需要读取值而不需要设置值，可以省略 set 块，定义一个只读下标：

// www.javascriptcn.com code example
struct MyArray {
    var elements: [Int]
    
    subscript(index: Int) -> Int {
        return elements[index]
    }
}

let myArray = MyArray(elements: [1, 2, 3, 4])
print(myArray[2]) // 输出: 3

3. 多参数下标

下标可以接受多个参数，例如用于访问二维数组中的元素：

// www.javascriptcn.com code example
struct Matrix {
    var grid: [[Int]]
    
    subscript(row: Int, column: Int) -> Int {
        get {
            return grid[row][column]
        }
        set(newValue) {
            grid[row][column] = newValue
        }
    }
}

var matrix = Matrix(grid: [[1, 2], [3, 4]])
print(matrix[1, 1]) // 输出: 4
matrix[1, 1] = 10
print(matrix[1, 1]) // 输出: 10

4. 类型下标

Swift 5.1 引入了类型下标（Type Subscripts），允许你在类型本身上定义下标：

// www.javascriptcn.com code example
enum Planet: Int {
    case mercury = 1, venus, earth, mars, jupiter, saturn, uranus, neptune
    
    static subscript(n: Int) -> Planet {
        return Planet(rawValue: n)!
    }
}

let planet = Planet[3]
print(planet) // 输出: earth

5. 下标的注意事项

• 下标可以有多个参数，但参数类型必须唯一。
• 下标可以重载，即同一个类型可以有多个不同参数类型的下标。
• 下标的返回值类型可以是任意类型，包括可选类型。

通过这些方式，你可以在 Swift 中灵活地使用下标来访问和修改自定义类型中的数据。