首页 > 解决方案 > 快速添加任意两个数字元组

问题描述

我创建了一个函数来添加看起来像 (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)应该工作。

标签: iosswiftgenerics

解决方案


元组需要在编译时确定它们的元素数量,因此类似可变参数的函数将不起作用。+对于需要支持的每个元组大小,您需要为运算符添加覆盖:

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]

这种方法的注意事项:

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

推荐阅读