Swift 高阶函数

作者: 半心_忬 | 来源:发表于2019-07-25 11:56 被阅读0次

    高阶函数的定义:

    Wikipedia 中,是这么定义高阶函数(higher-order function)的,如果一个函数:

    • 接收一个或多个函数当作参数

    • 把一个函数当作返回值

    至少满足以上条件中的一个的函数,那么这个函数就被称作高阶函数。

    使用高阶函数进行函数式编程的优势

    • 简化代码

    • 使逻辑更加清晰

    • 当数据比较大的时候,高阶函数会比传统实现更快,因为它可以并行执行(如运行在多核上)

    高阶函数在Swift语言中有大量的使用场景,本篇分析 Swift 提供的如下几个高阶函数:mapflatMapcompactMapfilterreduce


    一、map

    map方法获取一个闭包表达式作为其唯一参数。

    数组中的每一个元素调用一次该闭包函数,并返回该元素所映射的值。

    简单说就是数组中每个元素通过某种规则(闭包实现)进行转换,最后返回一个新的数组。

    1. 基本使用

    a. 需求:将Int类型数组中的元素乘以2,然后转换为String类型的数组

    
    let ints = [1, 2, 3, 4]
    
    let strs = ints.map { "\($0 * 2)" }
    
    // 打印结果:["2", "4", "6", "8"]
    
    print(strs)
    
    

    b. 需求:生成一个新的Int数组,元素是多少元素就重复多少个

    
    let nums = [1, 2, 3, 4]
    
    let mapNums = nums.map { Array(repeating: $0, count: $0) }
    
    // 打印结果:[[1], [2, 2], [3, 3, 3], [4, 4, 4, 4]]
    
    print(mapNums)
    
    

    最终返回的是一个二维数组。

    c. 需求:将String类型的数组转换为Int类型的数组

    
    let someAry = ["12", "ad", "33", "cc", "22"]
    
    // var nilAry: [Int?]
    
    let nilAry = someAry.map { Int($0) }
    
    // 打印结果:[Optional(12), nil, Optional(33), nil, Optional(22)]
    
    print(nilAry)
    
    

    最终返回的是[Int?],一个可选类型的Int数组,且元素中存在nil

    2. 源码分析

    由于Swfit是开源的,所以我们可以通过源码来分析map具体做了些什么。

    源码:https://github.com/apple/swift/blob/master/stdlib/public/core/Collection.swift

    
    public func map<T>(
    
        _ transform: (Element) throws -> T
    
      ) rethrows -> [T] {
    
        // TODO: swift-3-indexing-model - review the following
    
        let n = self.count
    
        if n == 0 {
    
          return []
    
        }
    
        var result = ContiguousArray<T>()
    
        result.reserveCapacity(n)
    
        var i = self.startIndex
    
        for _ in 0..<n {
    
          result.append(try transform(self[i]))
    
          formIndex(after: &i)
    
        }
    
        _expectEnd(of: self, is: i)
    
        return Array(result)
    
    }
    
    

    对于这个代码,我们可以看出,它做了以下几件事情:

    1. 构造一个名为 result 且与原数组的 capacity 一致的新数组,用于存放新的结果;
    
    2. 遍历自己的元素,对于每个元素,调用闭包的转换函数 transform ,进行转换;
    
    3. 将转换的结果使用 append 方法放入 result 中;
    
    4. 遍历完成后,返回 result 。
    

    tips: ContiguousArraySwift提供的更高性能的数组,几乎与Array没什么区别,如果不涉及Objective-C的混编或需要转NSArray,完全可以使用ContiguousArray取代Array来使用,可以有更高的性能。

    至于与Array的区别,这里就不拓展了,有兴趣的小伙伴Google一下。

    二、flatMap

    
    // flatMap 定义
    
    public func flatMap<SegmentOfResult>(_ transform: (Element) throws -> SegmentOfResult) rethrows -> [SegmentOfResult.Element] where SegmentOfResult : Sequence
    
    // Swift 4.1 以前的定义,4.1之后改名为 compactMap,compactMap时会详细说明
    
    @available(swift, deprecated: 4.1, renamed: "compactMap(_:)", message: "Please use compactMap(_:) for the case where closure returns an optional value")
    
    public func flatMap<ElementOfResult>(_ transform: (Element) throws -> ElementOfResult?) rethrows -> [ElementOfResult]
    
    

    flatMap的实现与map非常类似,也是数组中每个元素通过某种规则(闭包实现)进行转换,最后返回一个新的数组。

    不过flatMap能把数组中存有数组的数组(二维数组、N维数组)一同打开变成一个新的数组,称为降维,通俗一点就是把多维数组都会拍扁为一维数组,通过后面的例子一看就明白了。

    flatMap另一个解包的功能在 4.1 版本之后更名为compactMap,所以在compactMap再做说明。

    1. 基本使用

    a. 需求:将Int类型数组中的元素乘以2,然后转换为String类型的数组

    
    let ints = [1, 2, 3, 4]
    
    let strs = ints.flatMap { "\($0 * 2)" }
    
    // flatMap打印结果:["2", "4", "6", "8"]
    
    print(strs)
    
    

    该例子使用的与map一样,结果也是一样的,普通情况下,两者的效果一致。

    b. 需求:生成一个新的Int数组,元素是多少元素就重复多少个

    
    let nums = [1, 2, 3, 4]
    
    let mapNums = nums.flatMap { Array(repeating: $0, count: $0) }
    
    // flatMap打印结果:[1, 2, 2, 3, 3, 3, 4, 4, 4, 4]
    
    print(mapNums)
    
    

    可以看到,flatMap把数组中的数组都打开了,最终返回的是一个一维数组。而map返回的是一个二维数组,没有降维。接下来通过源码,来分析两者的差别。

    2. 源码分析

    源码:https://github.com/apple/swift/blob/master/stdlib/public/core/SequenceAlgorithms.swift

    
    public func flatMap<SegmentOfResult: Sequence>(
    
        _ transform: (Element) throws -> SegmentOfResult
    
      ) rethrows -> [SegmentOfResult.Element] {
    
        var result: [SegmentOfResult.Element] = []
    
        for element in self {
    
          result.append(contentsOf: try transform(element))
    
        }
    
        return result
    
    }
    
    

    我们可以看出,它做了以下几件事情:

    1. 构造一个名为 result 的新数组,用于存放新的结果;
    
    2. 遍历自己的元素,对于每个元素,调用闭包的转换函数 transform ,进行转换;
    
    3. 将转换的结果使用 append-contentsOf 方法放入 result 中;
    
    4. 遍历完成后,返回 result 。
    

    仔细观察,flatMapmap是有一些区别的:

    A. transform的差别

    • maptransform接收的参数是数组元素然后输出的是闭包执行后的类型T,最终执行的结果的是[T]

    • flatMaptransform接收的参数是数组的元素,但输出的一个Sequence类型,最终执行的结果并不是Sequence的数组,而是Sequence内部元素另外组成的数组,即:[Sequence.Element]

    B. 第三个步骤的差别

    • map使用append方法放入result中,所以transform之后的结果是什么类型,就将什么类型放入result中;

    • flatMap使用append-contentsOf方法放入result中,而appendContentsOf方法就是把Sequence中的元素一一取出来,然后再放入result中,这也就是flatMap能降维的原因。

    tips: append-contentsOf方法是干什么用的,看一段代码就明白了:

    
    // 定义:
    
    public mutating func append<S>(contentsOf newElements: __owned S) where Element == S.Element, S : Sequence
    
    var arrCons = [1, 3, 2]
    
    arrCons.append(contentsOf: [4, 5])
    
    // 打印结果:[1, 3, 2, 4, 5]
    
    print(arrCons)
    
    

    也就是会把目标Sequence的元素一一取出来,然后放入定义好的数组。

    所以flatMap必须要求 transform 函数返回的是一个Sequence类型,因为append-contentsOf方法需要的是一个Sequence类型的参数。

    Sequence是什么呢 ?Sequence是一个协议,主要有两个参数,一个是Element,也即是 Sequence里的元素,另一个则是Iterator(迭代器),且自身的Element与迭代器的Element是要相同的where Self.Element == Self.Iterator.Element。迭代器是遵循IteratorProtocol协议的,而IteratorProtocol的核心是next()方法,这个方法在每次被调用时返回序列中的下一个值,对Sequence的遍历,实际上调用的就是迭代器的next()方法,当序列下一个值为空时,next()返回nil,也就意味着遍历结束。Sequence的高阶函数都是在Sequence的扩展中定义的。

    这里就不拓展了,有兴趣的小伙伴Google以及查看官方源码。

    说到这里,我们就可以解释如下代码了:

    
    let nums = [1, 2, 3, 4]
    
    let mapNums = nums.map { Array(repeating: $0, count: $0) }
    
    let flatMapNums = nums.flatMap { Array(repeating: $0, count: $0) }
    
    // map打印结果:[[1], [2, 2], [3, 3, 3], [4, 4, 4, 4]]
    
    print(mapNums)
    
    // flatMap打印结果:[1, 2, 2, 3, 3, 3, 4, 4, 4, 4]
    
    print(flatMapNums)
    
    

    transform执行的结果都是一样的,都是得到一个数组,差别在于map将数组直接放入result中,而flatMap将数组中的元素一一取出来,更准确的说是调用Sequence的迭代器next()方法,将元素一一取出来,然后再放入result中。

    然后说说第一个例子,用mapflatMap的效果是一样的,是因为flatMap闭包执行后输出的Sequence是一个String类型,与用maptransform输出的类型是一致的,而String是遵循Sequence协议的并且只有一位Character,所以结果是一样的,如果我们转换的是多位的String,会是什么效果呢?

    看如下的代码:

    
    let ints = [1, 2, 3, 4]
    
    let mapStrs = ints.map { "\($0 * 2)啥啥啥" }
    
    // map打印结果:["2啥啥啥", "4啥啥啥", "6啥啥啥", "8啥啥啥"]
    
    print(mapStrs)
    
    let flatMapStrs = ints.flatMap { "\($0 * 2)啥啥啥" }
    
    // flatMap打印结果:["2", "啥", "啥", "啥", "4", "啥", "啥", "啥", "6", "啥", "啥", "啥", "8", "啥", "啥", "啥"]
    
    print(flatMapStrs)
    
    // String 的迭代器测试,这也就解释了flatMap在调用`append-contentsOf`方法是,是将 transform 后得到的 String 的 每一个 Character 放入了result中
    
    var testStr = "test"
    
    var iterator = testStr.makeIterator()
    
    // Optional("t")
    
    print(iterator.next())
    
    // Optional("e")
    
    print(iterator.next())
    
    // Optional("s")
    
    print(iterator.next())
    
    // Optional("t")
    
    print(iterator.next())
    
    // nil
    
    print(iterator.next())
    
    

    transform执行的结果都是一样的,都是得到一个String,差别在于mapString直接放入result中,而flatMapString中的Character一一取出来,更准确的说是调用String的迭代器next()方法,将Character一一取出来,然后再放入result中。

    三、Optional 中的 map 和 flatMap

    上面所分析的都是数组的mapflatMap,不只是数组中有这两个高阶函数,Optional有这两个高阶函数。

    先看看Optional中的定义:

    
    public enum Optional<Wrapped> : ExpressibleByNilLiteral {
    
    
    
        case none
    
        case some(Wrapped)
    
    
    
        public init(_ some: Wrapped)
    
        public func map<U>(_ transform: (Wrapped) throws -> U) rethrows -> U?
    
        public func flatMap<U>(_ transform: (Wrapped) throws -> U?) rethrows -> U?
    
    }
    
    
    1. 基本使用

    a. 需求:修改一个可选Int的值

    
    let a1: Int? = 1
    
    let b1 = a1.map { $0 * 2 }
    
    // 打印结果:Optional(2)
    
    print(b1)
    
    let a2: Int? = nil
    
    let b2 = a2.map { $0 * 2 }
    
    // 打印结果:nil
    
    print(b2)
    
    

    通过例子,说明对于一个Optional的变量来说,map方法允许它再次修改自己的值,并且不必关心自己是否为.None

    b. 需求:对NSDate做format操作

    
    // 不使用 map 的写法
    
    let date: Date? = Date()
    
    let formatter = DateFormatter()
    
    formatter.dateFormat = "YYYY-MM-dd"
    
    var formatted: String? = nil
    
    if let d = date {
    
        formatted = formatter.string(from: d)
    
    }
    
    // 打印结果:Optional("2019-07-16")
    
    print(formatted)
    
    // 使用 map 函数后,代码变得更短,更易读:
    
    let date: Date? = Date()
    
    let formatter = DateFormatter()
    
    formatter.dateFormat = "YYYY-MM-dd"
    
    let formatted = date.map { formatter.string(from: $0) }
    
    // Optional("2019-07-24")
    
    print(formatted)
    
    

    通过例子,我们可以得出结论,当输入的是一个Optional,同时需要在逻辑中处理这个Optional是否为nil,那么就适合用map来替代原来的写法,使得代码更加简短。

    c. 需求:将一个字符串转换成Int

    
    let s: String? = "abc"
    
    let mapR = s.map { Int($0) }
    
    let flatMapR = s.flatMap { Int($0) }
    
    // Optional(nil) --> map 会多包一层Optional
    
    print(mapR)
    
    // nil
    
    print(flatMapR)
    
    

    从上面的例子,我们可以得出结论,当我们的闭包参数有可能返回nil的时候,使用flatMap会更加合适,map会多包一层Optional,这样就很容易导致多重Optional嵌套的问题。

    什么是多重Optional嵌套呢,Optionalmap使用不当,就会导致多重Optional嵌套的问题。

    我们来看一段代码:

    
    let tq: Int? = 1
    
    // let b: Int??
    
    let b = tq.map { (a: Int) -> Int? in
    
        if a % 2 == 0 {
    
            return a
    
        } else {
    
            return nil
    
        }
    
    }
    
    // 打印结果:"b is not nil"
    
    if let _ = b {
    
        print("b is not nil")
    
    } else {
    
        print("b is nil")
    
    }
    
    

    由上面的代码就是Optionalmap使用不当而导致的多重Optional嵌套,多重Optionalnilif-let的判定是失效的,所以在工作中尽量避免多重Optional嵌套问题。上面例子的解决办法就是将map替换成flatMap,由于flatMap会有一次解包操作,所以能避免多重Optional嵌套的问题。

    关于多重Optional嵌套的问题,这里就不拓展了,有兴趣的小伙伴可以看看这里:

    http://blog.devtang.com/2016/02/27/swift-gym-1-nested-optional/

    2. 源码分析

    源码:https://github.com/apple/swift/blob/master/stdlib/public/core/Optional.swift

    
    @inlinable
    
      public func map<U>(
    
        _ transform: (Wrapped) throws -> U
    
      ) rethrows -> U? {
    
        switch self {
    
        case .some(let y):
    
          return .some(try transform(y))
    
        case .none:
    
          return .none
    
        }
    
      }
    
    
    
      @inlinable
    
      public func flatMap<U>(
    
        _ transform: (Wrapped) throws -> U?
    
      ) rethrows -> U? {
    
        switch self {
    
        case .some(let y):
    
          return try transform(y)
    
        case .none:
    
          return .none
    
        }
    
      }
    
    

    这两函数是惊人的相似,不仔细看的话,甚至看不出这两个函数的差别,两函数实现当然是差别的:

    1. map 返回的是 U ,flatMap 返回的是 U?
    
    2. map 把结果放到 .Some 里面返回,也就是会调用一次 Optional 的构造函数,多包一层 Optional ,flatMap把结果直接返回
    

    两个函数最终都保证了返回结果是Optional的,只是将结果转换成Optional的位置不一样。

    既然OptionalmapflatMap本质上是一样的,为什么要搞两种形式呢?这其实是为了调用者更方便而设计的。调用者提供的闭包函数,既可以返回Optional的结果,也可以返回非Optional的结果。对于后者,使用map方法,即可以将结果继续转换成Optional的。结果是Optional的意味着我们可以继续链式调用,也更方便我们处理错误。

    最后我们来看一段代码:

    
    var arr = [1, 2, 4]
    
    let res = arr.first.flatMap {
    
        arr.reduce($0, max)
    
    }
    
    // 打印结果:Optional(4)
    
    print(res)
    
    

    代码的意思是:计算出数组中的元素最大值,按理说,求最大值直接使用reduce方法就可以了,不过有一种特殊情况需要考虑,即数组中的元素个数为 0 的情况,在这种情况下,没有最大值。

    所以这里使用OptionalflatMap方法来处理了这种情况。arrfirst方法返回的结果是Optional的,当数组为空的时候,first方法返回.None,所以,这段代码可以处理数组元素个数为 0 的情况了。

    
    var arr: [Int] = []
    
    let res = arr.first.flatMap {
    
        arr.reduce($0, max)
    
    }
    
    // nil
    
    print(res)
    
    

    四、compactMap

    compactMap是在4.1之后对flatMap的一个重载方法的重命名,同样是数组中每个元素通过某种规则(闭包实现)进行转换,最后返回一个新的数组,不过compactMap会将nil剔除,并对Optional进行解包。

    1. 基本使用

    a. 需求:将String类型的数组转换为Int类型的数组

    
    var someAry = ["12", "ad", "33", "cc", "22"]
    
    // var compactMapAry: [Int]
    
    var compactMapAry = someAry.compactMap { Int($0) }
    
    // compactMap打印结果:[12, 33, 22]
    
    print(compactMapAry)
    
    

    最终返回的是[Int],一个Int数组,并将其中转换失败的nil过滤掉了,并对转换成功的Optional值进行了解包。

    2. 源码分析

    源码:https://github.com/apple/swift/blob/master/stdlib/public/core/SequenceAlgorithms.swift

    
    public func compactMap<ElementOfResult>(
    
        _ transform: (Element) throws -> ElementOfResult?
    
      ) rethrows -> [ElementOfResult] {
    
        return try _compactMap(transform)
    
    }
    
    public func _compactMap<ElementOfResult>(
    
    _ transform: (Element) throws -> ElementOfResult?
    
    ) rethrows -> [ElementOfResult] {
    
        var result: [ElementOfResult] = []
    
        for element in self {
    
          if let newElement = try transform(element) {
    
            result.append(newElement)
    
          }
    
        }
    
        return result
    
    }
    
    

    通过代码,我们可以看出,它做了以下几件事情:

    1. 调用 _compactMap 方法 传入 transform;
    
    2. 构造一个名为 result 的新数组,用于存放新的结果;
    
    3. 遍历自己的元素,对于每个元素,调用闭包的转换函数 transform ,进行转换;
    
    4. 将转换的结果 使用 if - let 后,再使用 append 方法放入 result 中;
    
    5. 遍历完成后,返回 result 。
    

    从这里就可以得出结论,compactMapmap的区别就在于,maptransform后的结果直接放入result中,而compactMap使用if-let后再放入result中,而if-let的作用就是解包和过滤nil

    在看如下代码,其中的差别以及为什么会有差别,就很清晰了。

    
    var someAry = ["12", "ad", "33", "cc", "22"]
    
    var mapAry = someAry.map { Int($0) }
    
    var compactMapAry = someAry.compactMap { Int($0) }
    
    // map打印结果:[Optional(12), nil, Optional(33), nil, Optional(22)]
    
    print(mapAry)
    
    // compactMap打印结果:[12, 33, 22]
    
    print(compactMapAry)
    
    

    五、filter

    filter用来过滤元素,即筛选出数组元素中满足某种条件(闭包实现)的元素。

    1. 基本使用

    a. 需求:筛选出Int数组中的偶数

    
    let nums = [1, 13, 12, 36, 77, 89, 96]
    
    let result = nums.filter { $0 % 2 == 0 }
    
    // 打印结果:[12, 36, 96]
    
    print(result)
    
    

    最终返回的是全部是偶数的Int数组。

    2. 源码分析

    源码:https://github.com/apple/swift/blob/master/stdlib/public/core/Sequence.swift

    
    public __consuming func filter(
    
        _ isIncluded: (Element) throws -> Bool
    
      ) rethrows -> [Element] {
    
        return try _filter(isIncluded)
    
      }
    
      @_transparent
    
    public func _filter(
    
        _ isIncluded: (Element) throws -> Bool
    
      ) rethrows -> [Element] {
    
        var result = ContiguousArray<Element>()
    
        var iterator = self.makeIterator()
    
        while let element = iterator.next() {
    
          if try isIncluded(element) {
    
            result.append(element)
    
          }
    
        }
    
        return Array(result)
    
    }
    
    

    通过代码,我们可以看出,它做了以下几件事情:

    1. 调用 _filter 方法 传入 isIncluded ;
    
    2. 构造一个名为 result 的新数组,用于存放新的结果;
    
    3. 使用迭代器,遍历所有的元素,对于每个元素,调用闭包 isIncluded ,判断是否符合条件;
    
    4. 将符合条件的元素使用 append 方法放入 result 中;
    
    5. 遍历完成后,返回 result 。
    

    六、reduce

    reduce有两个函数,先看看定义:

    
    @inlinable public func reduce<Result>(_ initialResult: Result, _ nextPartialResult: (Result, Element) throws -> Result) rethrows -> Result
    
    @inlinable public func reduce<Result>(into initialResult: __owned Result, _ updateAccumulatingResult: (inout Result, Element) throws -> ()) rethrows -> Result
    
    

    reduce两个函数都是把数组元素组合计算为另一个值,并且会接受一个初始值,这个初始值的类型可以和数组元素类型不同,这样子就有很大的可操作空间,也会很有趣,后面会说到。

    reduce两个函数效果是一样的,但也是有差别的,差别就在于闭包的定义:

    • 第一个函数闭包,接收ResultElement,返回闭包执行后的Result,后续的操作是将每次闭包执行后的Result当做下一个元素执行闭包的入参,直到遍历完所有元素;

    • 第二个函数闭包,接收的依然是ResultElement,不过没有返回值,并且Result是用inout修饰的,所以传入闭包的是Result的地址,所以闭包的执行都是基于Result进行操作的,这么说可能有些抽象,下面的源码分析,一看就明白了。

    还有一个区别就是初始值,第二个函数使用了__owned进行了修饰,这个我也没懂是为什么,如果有知道的小伙伴望不吝赐教。

    这两个函数实现的效果是一样的,只是实现的方式不同而已,第一个函数能做到的事情,第二个函数都能做到。

    1. 基本使用

    a. 需求:求一个Int类型数组的和

    
    var arr = [2, 3, 4, 5]
    
    // 正经写法
    
    //let r = arr.reduce(0) { $0 + $1 }
    
    // 简写
    
    let r = arr.reduce(0, +)
    
    // 打印结果:14
    
    print(r)
    
    let r2 = arr.reduce(into: 0, +=)
    
    // 打印结果:14
    
    print(r2)
    
    

    最终返回的 14,即 0 + 2 + 3 + 4 + 5 的和,其中0是初始值。除了能获取和,当然也能能获取积、商和差。let r = arr.reduce(0, +) 简写中的符号,可以使用 + - * /,在/的时候注意除数不能为0,可以修改试试看。

    这里能够简写的原因是由于Swift强大的类型推导,闭包仅仅传入了一个+号,Swift推导过程是首先nextPartialResult闭包有两个传入参数ResultElement,除此之外别无其他,因此+只能对这两个参数求和,得到一个结果值称为x吧,由于nextPartialResult函数还需要返回一个结果值,但是除了x也没有其他的可能,因此把x作为闭包结果值返回和Result进行相加计算,然后返回。

    b. 需求:求出数组中奇数的和、以及偶数乘积

    
    let nums = [1, 3, 2, 4]
    
    typealias ResTuple = (Int, Int)
    
    let res = nums.reduce((0, 1)) { (r, i) -> ResTuple in
    
        var temp = r
    
        if i % 2 == 0 {
    
            temp.1 *= i
    
        } else {
    
            temp.0 += i
    
        }
    
        return temp
    
    }
    
    // 打印结果:奇数和:4,偶数乘积:8
    
    print("奇数和:\(res.0),偶数乘积:\(res.1)")
    
    let res1 = nums
    
        .reduce(into: (0, 1)) { $1 % 2 == 0 ? ($0.1 *= $1) : ($0.0 += $1) }
    
    // 打印结果:reduce-into --> 奇数和:4,偶数乘积:8
    
    print("reduce-into --> 奇数和:\(res1.0),偶数乘积:\(res1.1)")
    
    

    通过这个例子,reduce能做的不仅仅是求和或拼接,还可以做更多个性化的事情。上面实现中最有意思的莫过于我们使用Tuple作为initialResult。一旦你尝试将reduce进入到日常工作流中,会渐渐发现,Tuple是一个不错的选择,它能够将数据与reduce操作快速挂钩起来,后面的有趣例子还会用到,相当地契合场景。

    2. 源码分析

    源码:https://github.com/apple/swift/blob/master/stdlib/public/core/SequenceAlgorithms.swift

    reduce第一个函数的源码:

    
    @inlinable
    
    public func reduce<Result>(
    
    _ initialResult: Result,
    
    _ nextPartialResult:
    
      (_ partialResult: Result, Element) throws -> Result
    
    ) rethrows -> Result {
    
        var accumulator = initialResult
    
        for element in self {
    
          accumulator = try nextPartialResult(accumulator, element)
    
        }
    
        return accumulator
    
    }
    
    

    通过以上代码代码,我们可以看出,它做了以下几件事情:

    1. 定义 accumulator 临时变量,并赋值 initialResult ;
    
    2. 遍历所有的元素,对于每个元素,调用闭包 nextPartialResult;
    
    3. 将闭包执行的结果赋值给临时变量 accumulator ;
    
    4. 遍历完成后,返回 accumulator 。
    

    reduce第二个函数的源码:

    
    @inlinable
    
    public func reduce<Result>(
    
    into initialResult: __owned Result,
    
    _ updateAccumulatingResult:
    
        (_ partialResult: inout Result, Element) throws -> ()
    
      ) rethrows -> Result {
    
        var accumulator = initialResult
    
        for element in self {
    
          try updateAccumulatingResult(&accumulator, element)
    
        }
    
        return accumulator
    
    }
    
    

    通过以上代码代码,我们可以看出,它做了以下几件事情:

    1. 定义 accumulator 临时变量,并赋值 initialResult ;
    
    2. 遍历所有的元素,对于每个元素,调用闭包 updateAccumulatingResult,参数是临时变量 accumulator 的地址,闭包执行其实就是更新 accumulator 的值;
    
    3. 遍历完成后,返回 accumulator 。
    

    从以上源码中就能清晰地看出这两个函数的区别:

    • 第一个函数是将闭包执行的结果赋值给临时变量accumulator,然后遍历下一个元素,知道遍历结束,返回accumulator

    • 第二个函数是将临时变量accumulator地址当做闭包的第一个参数,闭包的执行就是在操作accumulator的值。

    所以说第一个函数能做到的事情,第二个函数都可以做得到,在有的时候,使用第二个函数,代码还能更简洁。这就看具体的场景,选择使用哪个函数实现。

    3. reduce 有趣的拓展

    reducemapflatMapcompactMapfilter的一种扩展的形式(后四个函数能干嘛,reduce就能用另外一种方式实现)。reduce的基础思想是将一个序列转换为一个不同类型的数据,期间通过一个累加器accumulator来持续记录递增状态。为了实现这个方法,我们会向reduce方法中传入一个用于处理序列中每个元素的结合闭包nextPartialResult

    A. 使用reduce实现map的功能

    
    let arr = [1, 3, 2]
    
    let r1 = arr.reduce([]) { $0 + [$1 * 2] }
    
    // 打印结果:[2, 6, 4]
    
    print(r1)
    
    let r2 = arr.reduce([]) {
    
        var temp = $0
    
        temp.append($1 * 2)
    
        return temp
    
    }
    
    // 打印结果:[2, 6, 4]
    
    print(r2)
    
    let r3 = arr.reduce(into: []) { $0 += [$1 * 2] }
    
    // reduce-into打印结果:[2, 6, 4]
    
    print(r3)
    
    

    这里提供了两种写法,第一种更为简洁,第二种显得不那么简洁,但是第一种的效率是要比第二种低的,[2, 6] + [4]执行速度要慢于[2, 6].append(4)。倘若在处理庞大的列表,应取代集合 + 集合的方式,转而使用一个可变的 accumulator 变量进行递增。

    关于[2, 6] + [4]执行速度要慢于[2, 6].append(4)的效率问题,这里不做拓展,有兴趣的小伙伴可以参考:

    https://airspeedvelocity.net/2015/08/03/arrays-linked-lists-and-performance/

    B. 使用reduce实现filter的功能

    
    let nums = [1, 2, 3, 4]
    
    let result = nums.reduce([]) { $1 % 2 == 0 ? $0 + [$1] : $0 }
    
    // 打印结果:[2, 4]
    
    print(result)
    
    let r2 = nums.reduce(into: []) { $0 += $1 % 2 == 0 ? [$1] : [] }
    
    // reduce-into打印结果:[2, 4]
    
    print(r2)
    
    

    C. 使用reduce实现flatMap的功能

    
    let nums = [1, 2, 3, 4]
    
    let reduceNums = nums.reduce([]) { $0 + Array(repeating: $1, count: $1) }
    
    // reduce打印结果:[1, 2, 2, 3, 3, 3, 4, 4, 4, 4]
    
    print(reduceNums)
    
    let r2 = nums.reduce(into: []) { $0 += Array(repeating: $1, count: $1) }
    
    // reduce-into打印结果:[1, 2, 2, 3, 3, 3, 4, 4, 4, 4]
    
    print(r2)
    
    

    D. 使用reduce实现compactMap的功能

    
    var someAry = ["12", "ad", "33", "cc", "22"]
    
    // var reduceAry: [Int]
    
    var reduceAry = someAry.reduce([Int]()) {
    
        if let i = Int($1) {
    
            return $0 + [i]
    
        }
    
        return $0
    
    }
    
    // reduce打印结果:[12, 33, 22]
    
    print(reduceAry)
    
    var r2Ary = someAry.reduce(into: [Int]()) {
    
        if let i = Int($1) {
    
            $0 += [i]
    
        }
    
    }
    
    // reduce-into打印结果:[12, 33, 22]
    
    print(r2Ary)
    
    

    这里只做演示,使用的是最简洁的代码,实际项目中,还是建议使用效率更高的方式。

    从这里看来,reduce能做的这些系统已经提供了更好的实现方式,而且性能比reduce来实现要高很多,写这四段代码,只是让我们更深入地理解reduce的实现方式以及它的灵活性,reduce能做的其实更多。

    比如: 基于某个标准对一个Int数组做划分,区分奇数和偶数的数组

    
    let nums = [1, 2, 3, 4, 5, 6, 7, 8, 9]
    
    typealias ResTuple = ([Int], [Int])
    
    let res = nums.reduce(([], [])) { (r, i) -> ResTuple in
    
        if i % 2 == 0 {
    
            return (r.0, r.1 + [i])
    
        } else {
    
            return (r.0 + [i], r.1)
    
        }
    
    }
    
    // 打印结果:奇数:[1, 3, 5, 7, 9],偶数:[2, 4, 6, 8]
    
    print("奇数:\(res.0),偶数:\(res.1)")
    
    // 留给读者:使用reduce-into应该怎么写呢?
    
    

    如果使用常规的思路的话,会怎么实现呢?使用reduce是不是代码更加简洁,逻辑更加清晰。

    reduce除了较强的灵活性之外,还具有另一个优势:通常情况下,mapfilter所组成的链式结构会引入性能上的问题,因为它们需要多次遍历你的集合才能最终得到结果值,这种操作往往伴随着性能损失,比如以下代码:

    
    [0, 1, 2, 3, 4]
    
        .map({ $0 + 3})
    
        .filter({ $0 % 2 == 0})
    
        .reduce(0, +)
    
    

    初始序列(即[0, 1, 2, 3, 4])被重复访问了三次之多。首先是map,接着 filter,最后对数组内容求和,对于这种实现方式,实际是浪费了CPU的性能的,如果使用reduce,可以完美替代,且极大提高执行效率:

    
    [0, 1, 2, 3, 4]
    
        .reduce(0) { ($0 + 3) % 2 == 0 ? $1 + $0 + 3 : $1 }
    
    

    reduce的实现方式,只需要遍历 1 次就够了,代码也更加简洁。

    另外还有一些有趣的例子,或许对于reduce的使用打开思路有所帮助:

    A: 返回数组中有多少个不相同的数

    
    let nums = [11, 2, 3, 4, 5, 6, 5, 1]
    
    // let r: (Int?, Int)
    
    let r = nums.sorted(by: <)
    
            .reduce((.none, 0)) { ($1, $0.0 == $1 ? $0.1 : $0.1 + 1) }
    
    // 打印结果:7
    
    print(r.1)
    
    // 留给读者:使用reduce-into应该怎么写呢?
    
    

    B: 返回原数组分解成长度为 n 后的多个数组

    
    func chunk<T>(list: [T], length: Int) -> [[T]] {
    
        typealias Acc = (stack: [[T]], cur: [T], cnt: Int)
    
        let l = list.reduce((stack: [], cur: [], cnt: 0)) { (ac, o) -> Acc in
    
            if ac.cnt == length {
    
                return (stack: ac.stack + [ac.cur], cur: [o], cnt: 1)
    
            } else {
    
                return (stack: ac.stack, cur: ac.cur + [o], cnt: ac.cnt + 1)
    
            }
    
        }
    
        return l.stack + [l.cur]
    
    }
    
    // 打印结果:[[1, 2], [3, 4], [5, 6], [7]]
    
    print(chunk(list: [1, 2, 3, 4, 5, 6, 7], length: 2))
    
    // 留给读者:使用reduce-into应该怎么写呢?
    
    

    C: 给定一个 items 数组,每隔 count 个元素插入 element 元素,返回结果值,且需要确保 element 仅在中间插入,而不会添加到数组尾部

    
    func interpose<T>(items: [T], element: T, count: Int = 1) -> [T] {
    
        // cur 为当前遍历元素的索引值 cnt 为计数器,当值等于 count 时又重新置 1
    
        typealias Acc = (ac: [T], cur: Int, cnt: Int)
    
        return items.reduce((ac: [], cur: 0, cnt: 1), { (a, o) -> Acc in
    
            switch a {
    
            // 此时遍历的当前元素为序列中的最后一个元素
    
            case let (ac, cur, _) where cur + 1 == items.count:
    
                return (ac + [o], 0, 0)
    
            // 满足插入条件
    
            case let (ac, cur, c) where c == count:
    
                return (ac + [o, element], cur + 1, 1)
    
            case let (ac, cur, c):
    
                return (ac + [o], cur + 1, c + 1)
    
            }
    
        }).ac
    
    }
    
    // 打印结果:[1, 9, 2, 9, 3, 9, 4, 9, 5]
    
    print(interpose(items: [1, 2, 3, 4, 5], element: 9))
    
    // 打印结果:[1, 2, 9, 3, 4, 9, 5]
    
    print(interpose(items: [1, 2, 3, 4, 5], element: 9, count: 2))
    
    // 留给读者:使用reduce-into应该怎么写呢?
    
    

    D: 计算来自浙江的考生的高考平均分

    
    let students: [[String: String]] = [["name": "张三", "city": "杭州, ZJ", "score": "580"],
    
                                      ["name": "李四", "city": "南昌, JX", "score": "520"],
    
                                      ["name": "王五", "city": "长沙, HN", "score": "536"],
    
                                      ["name": "赵六", "city": "绍兴, ZJ", "score": "602"],
    
                                      ["name": "周七", "city": "宁波, ZJ", "score": "599"],
    
                                      ["name": "吴八", "score": "499"]] // 由于失误,城市丢失了
    
    typealias ResTuple = (cnt: Int, scoreTotal: Int)
    
    let restlt = students.reduce((0, 0)) { (r, s) -> ResTuple in
    
    
    
        // 如果城市缺失,或不是浙江的,或分数缺失则返回
    
        guard let city = s["city"],
    
            city.hasSuffix("ZJ"),
    
            let score = Int(s["score"] ?? "") else {
    
            return r
    
        }
    
        return (r.0 + 1, r.1 + score)
    
    }
    
    // 打印结果:浙江考生的平局分为:593.6667
    
    print("浙江考生的平局分为:\(Float(restlt.1) / Float(restlt.0))")
    
    // 留给读者:使用reduce-into应该怎么写呢?
    
    

    这些例子,算是抛砖引玉,希望对大家对reduce的理解和使用有所帮助。

    --

    参考文档:

    http://blog.devtang.com/2016/03/05/swift-gym-4-map-and-flatmap/

    http://www.hangge.com/blog/cache/detail_1827.html

    https://www.jianshu.com/p/56c99d31f2df

    https://www.jianshu.com/p/06c90c0470b2

    https://blog.csdn.net/offbye/article/details/50856101

    https://blog.csdn.net/offbye/article/details/50856101

    相关文章

      网友评论

        本文标题:Swift 高阶函数

        本文链接:https://www.haomeiwen.com/subject/spocrctx.html