Tantan Fu

Apr 04, 2016

函数式编程之 map

我们常常需要对一个数组中的每一个元素进行相应的转换,例如,下面这个函数将数组中的每个元素增加 1.
func incrementArray(xs: [Int]) -> [Int] { var result: [Int] = [] for x in xs { result.append(x + 1) } return result }
然而仔细考虑之后我们会发现,如果有其他的转化操作,就需要再重新写一个函数,而其他的代码逻辑都是相同的,所以,这里我们可以将转化部分的逻辑抽取成一个闭包,如下:
func doSomeThingOnArray(xs: [Int], transform: Int -> Int) -> [Int] { var result: [Int] = [] for x in xs { result.append(transform(x)) } return result }
这样,我们就将会变化的代码抽取出来了,例如将所有元素乘 2:
doSomeThingOnArray([1, 2, 3]) { (element) -> Int in return element * 2 }
然而代码写到这里,并不是一个很好的版本,因为这里的类型的信息是硬编码的,所以,我们来解决这个问题,并且将代码放到 Array 的 extension 中:
extension Array { func map<T>(transform: Element -> T) -> [T] { var result: [T] = [] for x in self { result.append(transform(x)) } return result } }
于是我们就可以像下面这样使用我们自己定义的 map:
[1, 2, 3].map { (x) -> Int in return x + 1 }
当然,Swift 标准库中提供了 map 的实现,定义在 SequenceType 协议中。
参考资料 objc Functional Swift https://www.futantan.com/blog/functional-programming-map

Copyright © 2022 Tantan Fu

Powered by noto.so