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

양의 정수 n이 매개변수로 주어질 때 n이 홀수면 n 이하의 홀수인 모든 양의 정수의 합을 리턴하고, n이 짝수면 n 이하의 짝수인 모든 양의 정수의 제곱의 합을 리턴하는 함수를 작성하라

 

 

어떤 숫자를 2로 나눴을 때 1이 남는다면 그 숫자는 홀수고, 1이 남지 않으면 그 숫자는 짝수다.

그리고 어떤 숫자 이하의 홀수 / 짝수는 모두 2를 빼야 얻을 수 있다. 예를 들어 10 이하의 짝수는 8, 6, 4, 2로 10에서 -2를 계속 하다보면 얻을 수 있다. 마찬가지로 7 이하의 홀수는 5, 3, 1로 역시 7에서 -2를 계속 하면 얻을 수 있다. 이제 어떻게 홀짝인 숫자들을 얻을 수 있을지 알았다.

그리고 홀수면 모든 양의 정수의 합을 리턴하고, 짝수면 모든 양의 정수의 합을 제곱해서 리턴해야 한다. 주먹구구식으로 풀면 아래와 같다.

 

class Solution {
    fun solution(n: Int): Int {
        var a = 0
        for (i in n downTo 1 step 2) {
            a += if (n % 2 == 1) {
                i
            } else {
                (i * i)
            }
        }

        return a
    }
}

 

코틀린이 제공하는 키워드인 downTo와 step은 값을 일정 수치만큼 감소시키면서 반복할 때 사용할 수 있는 키워드다.

만약 n이 10이라면 1이 나올 때까지 -2씩 줄어들며 for문이 작동한다. 그래서 8, 6, 4, 2를 각각 제곱해서 더한 다음 리턴한다. 홀수인 경우도 비슷하게 작동한다.

 

좀 더 간결하게 푼다면 아래와 같다.

 

class Solution {
    fun solution(n: Int): Int = if(n % 2 == 0) {
        (n downTo 1 step 2).sumOf { it * it }
    } else {
        (n downTo 1 step 2).sum()
    }
}