[프로그래머스 | JAVA] 주사위 게임 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	result
2	2	2	2	2222
4	1	4	4	1681
6	3	3	6	27
2	5	2	6	30
6	4	2	5	2

입출력 예 설명

- 입출력 예 #1
예제 1번에서 네 주사위 숫자가 모두 2로 같으므로 1111 × 2 = 2222점을 얻습니다. 따라서 2222를 return 합니다.
- 입출력 예 #2
예제 2번에서 세 주사위에서 나온 숫자가 4로 같고 나머지 다른 주사위에서 나온 숫자가 1이므로 (10 × 4 + 1)2 = 412 = 1681점을 얻습니다. 따라서 1681을 return 합니다.
- 입출력 예 #3
예제 3번에서 a, d는 6으로, b, c는 3으로 각각 같으므로 (6 + 3) × |6 - 3| = 9 × 3 = 27점을 얻습니다. 따라서 27을 return 합니다.
- 입출력 예 #4
예제 4번에서 두 주사위에서 2가 나오고 나머지 다른 두 주사위에서 각각 5, 6이 나왔으므로 5 × 6 = 30점을 얻습니다. 따라서 30을 return 합니다.
입출력 예 #5
예제 5번에서 네 주사위 숫자가 모두 다르고 나온 숫자 중 가장 작은 숫자가 2이므로 2점을 얻습니다. 따라서 2를 return 합니다.

 

제출답안 및 설명

import java.util.*;

class Solution {
    public int solution(int a, int b, int c, int d) {
        // 주사위 숫자의 출현 횟수를 저장할 HashMap 생성
        Map<Integer, Integer> scoreMap = new HashMap<>();
        
        // 각 주사위 값을 Map에 저장하면서 출현 횟수 계산
        scoreMap.put(a, 1);  // 첫 번째 숫자는 무조건 1회
        scoreMap.put(b, getInputValue(b, scoreMap));  // 나머지 숫자들은 기존 출현 횟수 확인 후 추가
        scoreMap.put(c, getInputValue(c, scoreMap));
        scoreMap.put(d, getInputValue(d, scoreMap));
        
        // 서로 다른 숫자의 개수에 따라 점수 계산
        int mapSize = scoreMap.size();
        if (mapSize == 1) {
            // 모든 주사위가 같은 경우 (p,p,p,p): 1111 × p
            return 1111 * a;
        } else if (mapSize == 2) {  
            if (scoreMap.containsValue(3)) {
                // 세 주사위가 같은 경우 (p,p,p,q): (10p + q)²
                int tripleKey = 0;  // 3번 나온 숫자
                int diffKey = 0;    // 1번 나온 숫자
                for (int key : scoreMap.keySet()) {
                    int value = scoreMap.get(key);
                    if (value == 3) {
                        tripleKey = key;
                    } else {
                        diffKey = key;
                    }
                }
                return (10 * tripleKey + diffKey) * (10 * tripleKey + diffKey);
            } else {
                // 두 주사위씩 같은 경우 (p,p,q,q): (p + q) × |p - q|
                int p = 0;
                int q = 0;
                int index = 0;
                for (int key : scoreMap.keySet()) {
                    if (index == 0) {
                        p = key;
                    } else {
                        q = key;
                    }
                    index++;
                }
                return (p + q) * Math.abs(p - q);
            }
        } else if (mapSize == 3) {
            // 두 주사위만 같고 나머지는 다른 경우 (p,p,q,r): q × r
            int q = 0;  // 한 번만 나온 첫 번째 숫자
            int r = 0;  // 한 번만 나온 두 번째 숫자
            int index = 0;
            for (int key : scoreMap.keySet()) {
                int value = scoreMap.get(key);
                if (value == 2) {   // 두 번 나온 숫자는 건너뜀
                    continue;
                }
                if (index == 0) {
                    q = key;
                } else {
                    r = key;
                }
                index++;
            }
            return q * r;
        } else {
            // 모든 주사위가 다른 경우: 최솟값 반환
            int[] compareArr = new int[scoreMap.size()];
            int index = 0;
            for (int key : scoreMap.keySet()) {
                compareArr[index] = key;
                index++;
            }
            Arrays.sort(compareArr);  // 오름차순 정렬
            return compareArr[0];     // 최솟값 반환
        }
    }
    
    // 주사위 숫자의 출현 횟수를 계산하는 헬퍼 메소드
    private int getInputValue(int num, Map<Integer, Integer> scoreMap) {
        if (scoreMap.containsKey(num)) {
            int value = (int) scoreMap.get(num);
            return value + 1;  // 이미 있는 숫자면 횟수 증가
        } else {
            return 1;  // 처음 나온 숫자면 1 반환
        }
    }
}

 

 

너무 복잡하게 작성한 것 같아 GPT에게 물어봤더니 아래와 같은 간단한 방법이 있었다!

역시 사람은 머리가 좋아야된다는걸 오늘도 느낀다.

class Solution {
    public int solution(int a, int b, int c, int d) {
        // 주사위 값을 배열로 정렬
        int[] dice = {a, b, c, d};
        Arrays.sort(dice);
        
        // 연속된 같은 숫자의 개수를 확인
        if (dice[0] == dice[3]) {  // 모두 같은 경우
            return 1111 * dice[0];
        }
        
        if (dice[0] == dice[2]) {  // 앞의 3개가 같은 경우
            return (10 * dice[0] + dice[3]) * (10 * dice[0] + dice[3]);
        }
        
        if (dice[1] == dice[3]) {  // 뒤의 3개가 같은 경우
            return (10 * dice[1] + dice[0]) * (10 * dice[1] + dice[0]);
        }
        
        if (dice[0] == dice[1] && dice[2] == dice[3]) {  // 2개씩 같은 경우
            return (dice[0] + dice[2]) * Math.abs(dice[0] - dice[2]);
        }
        
        if (dice[0] == dice[1]) {  // 앞의 2개만 같은 경우
            return dice[2] * dice[3];
        }
        
        if (dice[1] == dice[2]) {  // 중간 2개가 같은 경우
            return dice[0] * dice[3];
        }
        
        if (dice[2] == dice[3]) {  // 뒤의 2개가 같은 경우
            return dice[0] * dice[1];
        }
        
        // 모두 다른 경우
        return dice[0];  // 최솟값 반환
    }
}