[프로그래머스/Java] Lv.0 주사위 게임 2

문제

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

• 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
• 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a2 + b2 + c2 )점을 얻습니다.
• 세 숫자가 모두 같다면 (a + b + c) × (a2 + b2 + c2 ) × (a3 + b3 + c3 )점을 얻습니다.

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

- 제한사항

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

풀이

처음엔 제곱식을 계산하기 위해 Math.pow()를 사용하려고 했는데, 쓰고 보니 매개변수로 double 타입의 데이터를 사용해야 한다는 게 기억났다.

기본으로 주어진 변수 int answer를 사용하지 않아도 되긴 하는데, 파씽도 덧셈도 복잡할 것 같아서 그냥 곱셈식으로 풀어 쓴 다음 변수에 대입했다.

class Solution {
    public int solution(int a, int b, int c) {
        int answer = 0;
        
        int add = a + b + c;
        int squared = (a*a) + (b*b) + (c*c);
        int cubed = (a*a*a) + (b*b*b) + (c*c*c);
        
        if(a != b && b != c && a != c){
            answer = add;
        }else if(a == b && b == c && a == c){
            answer = add * squared * cubed;
        }else{
            answer = add * squared;
        }
        
        return answer;
    }
}

결과