[프로그래머스 Lv.0] 주사위 게임1(Kotlin)

문제

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

프로그래머스

풀이

수학 관련 함수를 사용하기 위해서는 import kotlin.math.*를 해줘야 한다.

제곱은 pow()함수를 사용한다. *pow는 반환형이 Double이기에 자료형이 다를 경우 toDouble()를 이용하여 바꿔줘야 한다.

이때 answer은 Int이므로 계산 후에 toInt()를 해준다.

절대값 계산은 abs()함수를 사용한다.

코드

import kotlin.math.* 

class Solution {
    fun solution(a: Int, b: Int): Int {
        var answer: Int = 0
        
        if(a%2!=0 && b%2!=0)
            answer=(a.toDouble().pow(2)+b.toDouble().pow(2)).toInt()
        else if(a%2!=0 || b%2!=0)
            answer=2*(a+b)
        else
            answer=abs(a-b)
        
        return answer
    }
}