[알고리즘] Swift -프로그래머스 연습문제 #181882(조건에 맞게 수열 변환하기 1)

2023. 5. 24. 20:34ALGORITHM/Swift

728x90
반응형

문제링크

 

 

 

 

분석

정수 배열이 주어질 때

각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고

50보다 작은 홀수라면 2를 곱합니다.

 

 

풀이

import Foundation

func solution(_ arr:[Int]) -> [Int] {

    return arr.map {
        if $0 >= 50 && $0 % 2 == 0 {
            return $0 / 2
        }
        if $0 < 50 && $0 % 2 == 1 {
            return $0 * 2
        }
        return $0
    }
}

 

 

결과

print(solution(   [1, 2, 3, 100, 99, 98]        ))

 

 

 

 

 

 

 

728x90
반응형