Algorithm/프로그래머스

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

Sheep1sik 2024. 5. 30. 18:47
반응형

문제

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

코드

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를 활용하여 풀어보았다.
반응형