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

문제링크

https://school.programmers.co.kr/learn/courses/30/lessons/181835

프로그래머스

 

 

 

 

 

분석

정수 배열 arr 과 k가 주어질 때

k가 홀수하면 arr의 모든 원소에 k를 곱하고 k가 짝수라면 arr의 모든 원소에 k를 더한다.

이러한 변환을 마친 후의 결과를 리턴하는 문제이다.

 

 

풀이

import Foundation

func solution(_ arr:[Int], _ k:Int) -> [Int] {
    return k % 2 == 0 ? arr.map { $0 + k } : arr.map { $0 * k }
}

 

 

결과

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