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

문제

1부터 6까지 숫자가 적힌 주사위가 네 개 있습니다. 네 주사위를 굴렸을 때 나온 숫자에 따라 다음과 같은 점수를 얻습니다.

• 네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
• 세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10 × p + q)^2 점을 얻습니다.
• 주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q(p ≠ q)라고 한다면 (p + q) × |p - q|점을 얻습니다.
• 어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r)이라면 q × r점을 얻습니다.
• 네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자 만큼의 점수를 얻습니다.

네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

- 제한사항

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

풀이

아 어려웠습니다. 헤데이크.
우선 a, b, c, d 모두 동일한 범위의 값을 가지고 있었기 때문에 정수형 배열에 담은 다음 Arrays.sort()를 통해 정렬을 하면 값을 비교할 때 수월하겠다는 생각을 했다. 실제로 연산을 해보니 정렬만 해도 IF문으로 나누어야 하는 경우의 수 자체가 확 줄어드는 것을 알 수 있었다.

경우에 따라 값을 구하는 과정은 대부분 쉬웠는데, 세 값은 P고 하나만 q일 때 연산해 주는 건 생각할 시간이 좀 필요했다. ||를 이용한 조건문이다 보니, 0~2가 같은지 1~3이 같은지에 따라 p와 q의 값이 달라져서였다.

그래서 어차피 else if로 조건이 한 번 걸러진 상태니까 삼항 연산자로 처리하기로 했다. 제곱 처리는 Math.pow()를 사용한 뒤 (int)로 파씽했다.

import java.util.*;

class Solution {
    public int solution(int a, int b, int c, int d) {
        int answer = 0;
        
        int[] dice = {a,b,c,d};
        Arrays.sort(dice);
        
        if(dice[0] == dice[3]){
            answer = 1111 * dice[0];
        }else if(dice[0] == dice[2] || dice[1] == dice[3]){
            answer = dice[0] == dice[2] ? (int)Math.pow(10 * dice[0] + dice[3], 2) : (int)Math.pow(10 * dice[3] + dice[0], 2);
        }else if(dice[0] == dice[1] && dice[2] == dice[3]){
            answer = (dice[0] + dice[3]) * (dice[3] - dice[0]);
        }else if(dice[0] == dice[1]){
            answer = dice[2] * dice[3];
        }else if(dice[1] == dice[2]){
            answer = dice[0] * dice[3];
        }else if(dice[2] == dice[3]){
            answer = dice[0] * dice[1];
        }else{
            answer = dice[0];
        }
        
        return answer;
    }
}

결과