본문 바로가기

프로그래머스

Lv 0 주사위 게임 1

문제요약

1부터 6까지 숫자가 적힌 주사위 두 개가 있다.

주사위를 굴린 숫자를 각각 a,b 라고 할 때

a 와 b가 모두 홀수라면 a제곱 + b제곱 

a 와 b중 하나만 홀수라면 2 * ( a + b)

a와 b가 모두 홀수가 아니라면 | a - b | 

결과를 구하라

 

입출력

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

 

 

문제풀이

import java.util.*;
class Solution {
    public int solution(int a, int b) {
        int answer = 0;
        
        if(a % 2 == 1 && b % 2 == 1){
            answer = a * a + b * b ;
        }else if((a % 2 == 0 && b % 2 == 1) || (a % 2 == 1 && b % 2 == 0)){
            answer =  2 * ( a + b ) ;
        }else{
            answer = Math.abs(a - b); 
        }
        
        return answer;
    }
}

 

절대값은 Math.abs() 를 사용했다.