[iOS - Swift] 1. Higher order function (고차함수) 직접 구현 방법 - map, filter, reduce

1. Higher order function (고차함수) 직접 구현 방법 - map, filter, reduce

2. Higher order function (고차함수) 직접 구현 방법 - flatMap, compactMap

Map, Filter, Reduce

• map, filter, reduce는 higher order function (고차함수)
• higher order function이란? 다음 두 가지 중 하나에 해당
  • 하나 이상의 함수를 인자로 받는 것
  • 함수를 결과로 리턴하는것

Map

• 주어진 Element 하나하나씩 접근하여 클로저로 넘겨준 함수 (higher order function)에 맵핑하여 새로운 타입을 반환

[1,2,3].map {
  print($0)
}

• 구현 방법

extension Array {
  func myMap<T>(_ f: (Element) -> T) -> [T] {
    var ret = [T]()
    for x in self {
      ret.append(f(x))
    }
    return ret
  }
}

[1,2,3].myMap {
  print($0)
}

Filter

• 주어진 Element 하나하나씩 접근하여 클로저로 넘겨준 함수 (higher order function)에 맵핑하며, 이 함수에서 true를 반환하는 값만 필터링

[1,2,3].filter { $0 % 2 == 0 } // [2]

• 구현 방법

extension Array {
  func myFilter(_ f: (Element) -> Bool) -> [Element] {
    var ret = [Element]()
    for x in self where f(x) {
      ret.append(x)
    }
    return ret
  }
}

[1,2,3].myFilter { $0 % 2 == 0 } // [2]

Reduce

• 주어진 Element에 하나씩 접근하여 클로저로 넘겨준 함수 (higher order function)에 맵핑하여, 이 함수를 통해 내부의 값들을 결합하여 차원을 축소(reduce)하는 것
  • 단순히 차원을 축소하여 기존 원소 타입과 동일하게 변환할 경우 reduce(_:_:)
  • 원소 타입을 변환할 경우 reduce(into:_:)

// 반환 타입이 기존 배열안의 원소 타입과 동일 (Element)
[1,2,3].reduce(0, { $0 + $1 }) // 6

// 반환 타입이 into 파라미터에 삽입한 타입으로 변환
[1,2,3].reduce(into: "", { $0 += String($1) }) // 123

• reduce(_:_:) 구현 방법

extension Array {
  func myReduce(_ initVal: Element, _ f: (Element, Element) -> Element) -> Element {
    var ret = initVal
    for x in self {
      ret = f(ret, x)
    }
    return ret
  }
}

[1,2,3].myReduce(0, { $0 + $1 }) // 6

• reduce(into:_:) 구현 방법
  • 핵심 - 외부에서 주입받는 함수에 inout T를 놓고, 안에서 값이 변경되도록 설정

extension Array {
  func myReduce<T>(into: T, _ f: (inout T, Element) -> ()) -> T {
    var ret = into
    for x in self {
      f(&ret, x)
    }
    return ret
  }
}

print([1,2,3].myReduce(into: "", { $0 += String($1) })) // 123