[Swift] reduce 사용하기 | sum

Programmers 하샤드 수를 풀고나서, reduce를 이용한 풀이를 마주쳐버렸다. 

// Programmers 하샤드 수 풀이
func solution(_ x:Int) -> Bool {

    var sum_x = String(x).map{ Int(String($0))! }.reduce(0, +)

    return x % sum_x == 0

}

ㅇㅇ.. 나도 공부해줄게

reduce

컨테이너 내부의 값을 하나로 통합(연산)하여 return

* Swift에서는 Python과는 달리 배열의 총 합을 구하는 내장 함수 sum이 없다. 대신 reduce를 응용해주면 된다.

(1) reduce(_: _:)

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

• 클로저가 각 요소를 전달받아 연산한 후 클로저 실행을 위해 값을 다시 반환하여 컨테이너를 순환
• 첫 번째 인자로 초깃값을 지정해주어야한다.

let numbers: [Int] = [1, 2, 3]

var sum: Int = numbers.reduce(0, { (result: Int, element: Int) -> Int in
    print("\(result) + \(element)")
    // 0(=초기값) + 1
    // 1(=0+1) + 2
    // 3(=1+2) + 3
    return result + element // 순환
})

print(sum)  // 6

// Closure 표현식을 사용하여 간결하게 표현
var sum2: Int = numbers.reduce(0) {
    print("\($0) + \($1)")
  // 0 + 1
  // 1 + 2
  // 3 + 3
    return $0 + $1
}
print(sum2) // 6

 

reduce의 첫번째 인자 0을 지정하며, result 내지 $0로 결과값을 return 받아 컨테이너 내부를 순환하는 것을 확인할 수 있다.

 

(2) reduce(into:_:)

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

• 컨테이너를 순환하여 클로저가 실행되지만 클로저가 따로 결과값을 반환하지 않음. 대신 inout 매개변수를 사용하여 초깃값에 직접 연산 실행

let numbers: [Int] = [1, 2, 3]

var sum = numbers.reduce(into: 0, { (result: inout Int, element: Int) in
    // 0 + 1
    // 1 + 2
    // 3 + 3
    print("\(result) + \(element)")
    result += element // !!클로저 값을 반환하지 않음!!
})
print(sum)  // 6

 

 

map vs reduce

: 같은 실행 결과를 다른 코드로 표현해보기

• reduce(into:_:)로 color 값을 대문자로 변환

let colors: [String] = ["orange", "yellow", "green", "blue", "purple"]

var rainbowColors: [String] = colors.reduce(into: [], {
  $0.append($1.uppercased())
})

print(rainbowColors) // ["ORANGE", "YELLOW", "GREEN", "BLUE", "PURPLE"]

• map으로 color 값을 대문자로 변환

let colors: [String] = ["orange", "yellow", "green", "blue", "purple"]

var rainbowColors: [String] = colors.map { $0.uppercased() }

print(rainbowColors) // ["ORANGE", "YELLOW", "GREEN", "BLUE", "PURPLE"]