[Lv. 0] 주사위 게임 1

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181839

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

1부터 6까지 숫자가 적힌 주사위가 두 개 있습니다. 두 주사위를 굴렸을 때 나온 숫자를 각각 a, b라고 했을 때 얻는 점수는 다음과 같습니다.

• a와 b가 모두 홀수라면 a2 + b2 점을 얻습니다.
• a와 b 중 하나만 홀수라면 2 × (a + b) 점을 얻습니다.
• a와 b 모두 홀수가 아니라면 |a - b| 점을 얻습니다.

두 정수 a와 b가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

 

제한 사항 : 

• a와 b는 1 이상 6 이하의 정수입니다.

입출력 예 :

a	b	result
3	5	35
6	1	14
2	4	2

풀이 : 

import kotlin.math.abs
import kotlin.math.pow

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