[Algorithm] 프로그래머스 - 주사위 게임 1 (Kotlin)

 

조건에 따라서 계산식을 작성하면 된다.

절대값을 구하는 abs()를 사용하려면 import문이 필요하니 주의한다.

 

class Solution {
    fun solution(a: Int, b: Int): Int {
        return when {
            a % 2 != 0 && b % 2 != 0 -> a * a + b * b
            (a % 2 != 0 || b % 2 != 0) -> 2 * (a + b)
            else -> kotlin.math.abs(a - b)
        }
    }
}