[Java] 주사위 게임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이하의 정수입니다.

입출력 예

문제 풀이

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

이렇게 작성하고 테스트 통과했길래 제출하니 실패했다.
실패한 이유는 a와 b가 다르고, b와 c가 다른 것만 체크했기 때문이다. a와 c가 다른지도 체크해줘야된다. 세 숫자가 모두 다를 때만 답이 a + b + c 가 되어야 한다.

수정한 코드이다.

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