Swift 下标脚本

下标脚本是一种特殊的访问器，允许通过索引的方式访问集合、列表或序列中的元素。你可以为自定义类或结构体定义下标脚本，使得它们的行为类似于数组、字典或其他标准集合类型。

定义和使用下标脚本

在 Swift 中，你可以为一个类或结构体定义一个下标脚本，以便能够通过索引值来访问实例中的某个特定值。下标脚本通常被用来实现快速、简洁地访问集合、列表或序列中的数据。

简单示例

假设我们有一个简单的类 TimesTable，用于表示一个乘法表：

// www.javascriptcn.com code example
struct TimesTable {
    let multiplier: Int

    subscript(index: Int) -> Int {
        return multiplier * index
    }
}

let threeTimesTable = TimesTable(multiplier: 3)
print(threeTimesTable[5]) // 输出 "15"

在这个例子中，subscript 关键字用于定义一个下标脚本，它接受一个整数参数 index 并返回一个整数。当通过 threeTimesTable[5] 访问时，实际上是调用了下标脚本，并将结果返回。

多参数下标脚本

除了单一参数外，你还可以定义具有多个参数的下标脚本，这可以让你通过多个索引值来访问元素。例如，可以定义一个二维数组的下标脚本：

// www.javascriptcn.com code example
class Matrix {
    let rows: Int, columns: Int
    var grid: [Double]

    init(rows: Int, columns: Int) {
        self.rows = rows
        self.columns = columns
        grid = Array(repeating: 0.0, count: rows * columns)
    }

    subscript(row: Int, column: Int) -> Double {
        get {
            return grid[(row * columns) + column]
        }
        set {
            grid[(row * columns) + column] = newValue
        }
    }
}

var matrix = Matrix(rows: 2, columns: 2)
matrix[0, 1] = 1.5
print(matrix[0, 1]) // 输出 "1.5"

在这个例子中，Matrix 类定义了一个二维数组的下标脚本。通过 matrix[row, column] 的方式可以设置或获取特定位置的元素。

下标脚本的规则和最佳实践

• 只读下标脚本：如果你的下标脚本只需要提供读取功能，那么可以省略 set 部分。

  // www.javascriptcn.com code example
  struct SimpleDictionary {
      private var storage: [String: Int] = [:]
  
      subscript(key: String) -> Int? {
          get {
              return storage[key]
          }
      }
  }
  
  var dict = SimpleDictionary()
  dict["key"] = 42
  print(dict["key"]) // 输出 "Optional(42)"

• 下标脚本的重载：你可以在同一个类或结构体中定义多个下标脚本，但这些下标脚本必须有不同的参数列表。

  // www.javascriptcn.com code example
  class CustomArray<T> {
      private var elements: [T]
  
      init(elements: [T]) {
          self.elements = elements
      }
  
      subscript(index: Int) -> T {
          return elements[index]
      }
  
      subscript(range: Range<Int>) -> [T] {
          return Array(elements[range])
      }
  }
  
  let customArray = CustomArray(elements: [1, 2, 3, 4, 5])
  print(customArray[2]) // 输出 "3"
  print(customArray[1..<4]) // 输出 "[2, 3, 4]"

• 避免过度依赖下标脚本：虽然下标脚本可以提供非常方便的数据访问方式，但在某些情况下，它可能会使代码难以理解和维护。因此，在设计你的 API 时，需要权衡是否使用下标脚本。

实践练习

为了更好地理解下标脚本的使用，尝试完成以下练习：

1. 创建一个名为 Bookshelf 的类，它可以存储书籍的名称和作者。使用下标脚本来获取和设置特定位置的书籍信息。
2. 扩展 Dictionary 类型，使其支持通过一个元组作为键来获取或设置值。例如，可以使用 (key1, key2) 形式的元组来访问嵌套字典。

通过这些练习，你将更熟练地掌握如何使用下标脚本，并将其应用于实际问题中。下标脚本是 Swift 中一种强大而灵活的功能，合理使用它可以大大简化你的代码并提高可读性。