Swift中的多种索引类型

Question

在Swift（一种我仍然比较陌生的语言）中，我试图定义一个允许使用Int或Range<Int>索引的类。例如：

var m = Matrix(rows: 10, cols: 10) 
var v = m[1, 0..<10] 

两种方式我能想到让这个问题的是，要么添加一个扩展范围类：

extension Range { 
    init(_ intValue:Element) { 
    self.init(start: intValue, end: intValue.successor()) 
    } 
} 

，或者创建一个枚举，允许两种类型：

enum IndexType { 
    case value(Int) 
    case range(Range<Int>) 

    init(_ v:Int) { 
     self = .value(v) 
    } 

    init(_ r:Range<Int>) { 
     self = .range(r) 
    } 
} 

但是，这两种解决方案都不能按预期工作，因为我仍然需要手动指定类型。例如，给定函数：

func slice(indices:IndexType...) -> [Double] { 
    // ... 
} 

我不能再只是做：

slice(1, 3...4, 5) 

，而是必须做的：

slice(IndexType(1), IndexType(3...4), 5) 

是否有做到这一点的方式迅速？我习惯于C++，它会自动进行类型推断，但是它似乎不适用于Swift。

一个类似的问题，我发现： Swift Arrays of Multiple Types

可是，我真的想避免使用Any，因为我知道这两种类型也是应该的。

Answer 1

protocol MatrixIndex { 
    var Matrix_range: Range<Int> { get } 
} 

extension Int : MatrixIndex { 
    var Matrix_range: Range<Int> { 
     get { 
      return Range<Int>(start: self, end: self+1) 
     } 
    } 
} 

extension Range : MatrixIndex { 
    var Matrix_range: Range<Int> { 
     get { 
      return Range<Int>(start: self.startIndex as! Int, end: self.endIndex as! Int) 
     } 
    } 
} 

class Matrix { 
    subscript(row: MatrixIndex, column: MatrixIndex) ->() { 
     get { 
      print("getting \(row.Matrix_range) \(column.Matrix_range)") 
     } 

     set(newValue) { 
      print("setting \(row.Matrix_range) \(column.Matrix_range)") 
     } 
    } 
}