[ 프로그래머스 ] 홀짝에 따라 다른 값 반환하기

문제

프로그래머스

 

코드

import Foundation

func solution(_ n:Int) -> Int {
    if n % 2 == 0 {
        return stride(from: 2, through: n, by: 2).reduce(0) { $0 + $1 * $1 }
    } else {
        return stride(from: 1, through: n, by: 2).reduce(0, +)
    }
}

 

💡풀이

stride 와 reduce를 활용하여 풀어보았다.