复制Array.reduce()方法

时间:2018-07-05 13:13:48

标签: swift closures higher-order-functions

我正在尝试在自定义类中复制Array.reduce()方法,并意识到它使用Result作为类型。只是不明白是将Result类型创建为Enum还是其他类型。

import Foundation

public class MyArray {
    private var arr: [Int] = []
    internal static var instance: MyArray?

    private init() {}

    public static func getInstance() -> MyArray {
        if self.instance == nil {
            self.instance = MyArray()
        }
        return self.instance!
    }

    public func insert(value val: Int) {
        arr.append(val)
    }

    /*************** Custom reduce like function ***************/

    public func perform(_ initialResult: Int, _ nextPartialResult: (Int, Int) -> Int) -> Int {
        var result = initialResult

        for element in arr {
            result = nextPartialResult(result, element)  // calling the closure
        }
        return result
    }
}

现在从外部访问MyArray类

var arr1 = MyArray.getInstance()
arr1.insert(value: 1)
arr1.insert(value: 2)
arr1.insert(value: 4)
arr1.insert(value: 3)
arr1.insert(value: 2)
arr1.insert(value: 5)
arr1.insert(value: 2)
arr1.insert(value: 2)

//  :Complex calculations left for user to implement
var result = arr1.perform(0) {
    return $0 + ( $1 * $1)
}
print("Complex calculation in elements of MEMBER array of arr1: \(result)")

//  :Just another way of writing the above closure
result = arr1.perform(0) { (result, num1) -> Int in
    return result + ( num1 * num1)
}
print("Complex calculation in elements of MEMBER array of hello arr1: \(result)")

//  :Simple calculations
print("Factorial of elements in MEMBER array of arr1: \(arr1.perform(1, *))")
print("Sum of elements in MEMBER array of arr1: \(arr1.perform(0, +))")

问题是我必须一次用一种特定类型(Int或String或Double等)定义我的perform()函数。我正在尝试创建我的函数以使其与reduce()函数一样可以使用任何类型。

我无法理解如何在我的课程中定义Result类型,然后在我的函数中使用它!

我知道Result类型不是swift标准库的一部分。

1 个答案:

答案 0 :(得分:1)

标准reduce函数使用泛型。请参阅Swift书中的Generics一章。

func reduce<Result>(_ initialResult: Result, _ nextPartialResult: (Result, Element) throws -> Result) rethrows -> Result

它具有两种通用类型:ResultElementElement来自集合中值的类型,Result来自于减少值的结果类型。

因此,第一步是在自己的perform函数中使用相同的签名。

但是这样做,您会发现现在需要使MyArray类也基于泛型,而不是被硬编码为仅与Int一起使用。

在尝试执行此操作时,您将发现无法将MyArray定义为通用的并且不能同时支持单例模式。因此,您需要删除instancegetIntance()

最终结果变为:

public class MyArray<Element> {
    private var arr: [Element] = []

    public init() {}

    public func insert(value val: Element) {
        arr.append(val)
    }

    /*************** Custom reduce like function ***************/

    public func perform<Result>(_ initialResult: Result, _ nextPartialResult: (Result, Element) -> Result) -> Result {
        var result = initialResult

        for element in arr {
            result = nextPartialResult(result, element)
        }

        return result
    }
}

有了这个,您的第一个示例将变为:

var arr1 = MyArray<Int>()
arr1.insert(value: 1)
arr1.insert(value: 2)
arr1.insert(value: 4)
arr1.insert(value: 3)
arr1.insert(value: 2)
arr1.insert(value: 5)
arr1.insert(value: 2)
arr1.insert(value: 2)

//  :Complex calculations left for user to implement
var result = arr1.perform(0) {
    return $0 + ( $1 * $1)
}
print(result)

这将输出期望的67结果。

最后,它可以工作,但是,如果您会注意到,这MyArray类没有意义(除了学习练习)。只需使用Array