快速添加任何两个数字元组

时间:2019-01-11 09:32:02

标签: ios swift generics

我创建了一个函数来添加看起来像(Int,Int)的元组。

func +<T : Numeric> (x: (T, T), y: (T, T)) -> (T, T) {
    return (x.0 + y.0, x.1 + y.1)
}

它适用于(10, 20) + (15, 15)

现在我需要做的是改进功能以接受相同长度的任何可变长度元组。

怎么可能?

最后(12) + (23)(10, 12, 16) + (11, 36, 25)应该可以工作。

2 个答案:

答案 0 :(得分:1)

Tuples需要在编译时确定其元素数量,因此类似可变参数的函数将不起作用。您需要为+运算符添加替代项,以适应需要支持的每个元组大小:

func +<T : Numeric> (x: (T, T), y: (T, T)) -> (T, T) {
    return (x.0 + y.0, x.1 + y.1)
}

func +<T : Numeric> (x: (T, T, T), y: (T, T, T)) -> (T, T, T) {
    return (x.0 + y.0, x.1 + y.1, x.2 + y.2)
}

func +<T : Numeric> (x: (T, T, T, T), y: (T, T, T, T)) -> (T, T, T, T) {
    return (x.0 + y.0, x.1 + y.1, x.2 + y.2, x.3 + y.3)
}

// and so on, ...

或者,您可以切换到其他数据类型,例如数组,它们允许动态数量的项目:

infix operator ++ 

func ++<T: Numeric>(_ lhs: [T], _ rhs: [T]) -> [T] {
    return zip(lhs, rhs).map { $0.0 + $0.1 }
}

print([10, 12, 16] ++ [11, 36, 25]) // [21, 48, 41]

此方法的要点:

  • 您需要使用其他运算符,因为已经为数组定义了+,并且它将数组连接在一起,而不是单独求和相应的元素
  • 如果两个数组的大小不同,则结果将为两个输入数组的较小数组的大小。

答案 1 :(得分:1)

您可以使用@Cristik建议的Array解决方案,也可以使用闭包返回可变参数的函数,例如:

func add<T : Numeric>(_ a: T...) -> (_ b: T...) -> [T] {
    return { (b: T...) -> [T] in
        return zip(a, b).map { $0.0 + $0.1 }
    }
}

let sum = add(1, 2,3)(4, 5, 6)

print(sum)