[Algorithm] 프로그래머스 - 조건에 맞게 수열 변환하기 1 (Kotlin)

정수 배열 arr이 주어진다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수면 2로 나누고 50보다 작은 홀수면 2를 곱한다. 그 결과인 정수 배열을 리턴하는 함수를 완성하라

 

 

문제 핵심은 2개다.

 

1. 숫자가 50보다 크거나 같은 짝수 : 2로 나눔
2. 숫자가 50보다 작거나 작은 홀수 : 2를 곱함

 

이 조건에 맞게 if 또는 when을 작성하면 된다. 그 외의 경우에는 num을 리턴하면 될 것이다.

 

class Solution {
    fun solution(arr: IntArray): IntArray = arr.map { num ->
        when {
            num >= 50 && num % 2 == 0 -> num / 2
            num < 50 && num % 2 != 0 -> num * 2
            else -> num
        }
    }.toIntArray()
}