Функции высшего порядка. Кастомная реализация. (Custom implementation of higher-order functions.)

Всем привет!

На интервью иногда просят реализовать одну из функций высшего порядка. Решил потренироваться в их самостоятельной реализации. Ниже представлен результат моих тренировок =)

Иногда бывает легче реализовать функцию конкретного типа, а затем сделать её универсальной (Generic). Такого подхода я и буду придерживаться в примерах, приведенных ниже.

Все функции написаны в расширении массива ( extension Array)

Reduce

extension Array {
    func customIntReduce(res: Int, handler:(Int, Int) -> Int) -> Int {
        var result = res
        for elem in self {
            result = handler(result, elem as! Int)
        }
        return result
    }
    
    func customReduce<T, U>(res: T, handler:(T, U) -> T) -> T {
        var result = res
        for elem in self {
            result = handler(result, elem as! U)
        }
        return result
    }
}

print([1, 2, 3].customIntReduce(res: 0) { $0 + $1 })
print([1, 2, 3].customReduce(res: 0) { $0 + $1 })

Filter

  func customIntFilter(handler: (Int) -> Bool) -> [Int] {
        var filteredArray: [Int] = []
        
        for element in self {
            if handler(element as! Int) {
                filteredArray.append(element as! Int)
            }
        }
        return filteredArray
    }
    
    func customFilter(handler: (Element) -> Bool) -> [Element] {
        var filteredArray: [Element] = []
        for element in self {
            if handler(element) {
                filteredArray.append(element)
            }
        }
        return filteredArray
    }

    print([1, 2, 3, 4, 5].customIntFilter{ $0 > 2})
    print([1, 2, 3, 4, 5].customFilter{ $0 > 2})

Map

 func customIntMap(handler: (Int) -> Any) -> [Any] {
        var mapedArray:[Any] = []
        for elem in self {
            let mapedValue = handler(elem as! Int)
            mapedArray.append(mapedValue)
        }
        return mapedArray
    }
    
    func customMap<T>(handler: (Element) -> T) -> [T] {
        var mapedArray: [T] = []
        for elem in self {
            let mapedValue = handler(elem)
            mapedArray.append(mapedValue)
        }
        return mapedArray
    }

    print([1, 2, 3].customIntMap{ "\($0 * $0)" })
    print([1, 2, 3].customMap{ "\($0 * $0)" })

Compact Map

func customStringCompactMap(_ handler: (String) -> Any?) -> [Any] {
        var mapedArray: [Any] = []
        for elem in self {
            if let mapedValue = handler(elem as! String) {
                mapedArray.append(mapedValue)
            }
        }
        return mapedArray
    }
    
    func customCompactMap<T>(handler: (Element) -> T?) -> [T] {
        var mapedArray: [T] = []
        for elem in self {
            if let mapedValue = handler(elem) {
                mapedArray.append(mapedValue)
            }
        }
        return mapedArray
    }

    print(["1", "two", "3"].customStringCompactMap{ Int($0) })
    print(["1", "two", "3"].customCompactMap{ Int($0) })

Flat Map

 func customIntFlatMap(handler: ([Int]) -> Any) -> [Any] {
        var mappedArray: [Any] = []
        forEach {
            mappedArray += handler($0 as! [Int]) as! Array<Any>
        }
        return mappedArray
    }


    func customFlatMap<T>(handler: (Element) -> [T]) -> [T] {
        var mappedArray: [T] = []
        forEach {
            mappedArray += handler($0)
        }
        return mappedArray
    }


    let arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
        
    let mapped = arr.customIntFlatMap{ $0 }
    print(mapped)
        
    let mapped2 = arr.customFlatMap{ $0 }
    print(mapped2)

Надеюсь примеры помогут кому-то лучше подготовиться к собеседованию, а кому-то лучше разобраться в дженериках.