[프로그래머스] 주사위 게임 3 (Java)

Posted: 2025.09.28 Updated: 2025.09.28

주사위 게임 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) {
        int answer = 0;
        int[] n = {a, b, c, d};
        
        Map<Integer, Integer> m = new HashMap<>();
        
        for (int i = 0; i < n.length; i++) {
            m.put(n[i], m.getOrDefault(n[i], 0) + 1);
        }
        
        int s = m.size();
        ArrayList<Integer> k = new ArrayList<>(m.keySet());
        
        if (s == 1) {
            answer = 1111 * n[0];
        }
        else if (s == 2) {
            if (m.get(k.get(0)) == m.get(k.get(1))) {
                int q = k.get(0);
                int p = k.get(1);
                
                answer = (p + q) * Math.abs(p - q);
            }
            else {
                if (m.get(k.get(0)) == 3) {
                    answer = (int) Math.pow(10 * k.get(0) + k.get(1), 2);
                }
                else {
                    answer = (int) Math.pow(10 * k.get(1) + k.get(0), 2);
                }
            }
        }
        else if (s == 3) {
            answer = 1;
            
            for (int t : k) {
                if (m.get(t) != 2) {
                    answer *= t;
                }
            }
        }
        else if (s == 4) {
            answer = 7;
            
            for (int i : n) {
                answer = Math.min(answer, i);
            }
        }
        
        return answer;
    }
}

풀이

중복된 숫자와 갯수를 파악하기 위해 HaspMap 을 사용했다.

Map<Integer, Integer> m = new HashMap<>();
        
for (int i = 0; i < n.length; i++) {
    m.put(n[i], m.getOrDefault(n[i], 0) + 1);
}

key는 숫자, value는 중복 횟수이다.

int s = m.size();
ArrayList<Integer> k = new ArrayList<>(m.keySet());

s 는 등장한 숫자의 총 갯수로, s 변수를 통해 조건을 분기한다.
k 는 중복 숫자를 제외한 숫자들의 배열이다.

if (s == 1) {
    answer = 1111 * n[0];
}

s 가 1일 경우, 모든 숫자가 같다는 뜻이므로 1111 * n[0] 를 반환한다.

else if (s == 2) {
    if (m.get(k.get(0)) == m.get(k.get(1))) {
        int q = k.get(0);
        int p = k.get(1);
                
        answer = (p + q) * Math.abs(p - q);
    }
    else {
        if (m.get(k.get(0)) == 3) {
            answer = (int) Math.pow(10 * k.get(0) + k.get(1), 2);
        }
        else {
            answer = (int) Math.pow(10 * k.get(1) + k.get(0), 2);
        }
    }
}

s 가 2일 경우, 주사위가 두 개씩 같은 값이 나온 경우와 같은 값이가 3번 나오고 한 숫자만 다른 경우로 구분된다.
이때, 맵에 계산해놓은 등장 횟수를 활용하여 분기한다.

else if (s == 3) {
    answer = 1;
            
    for (int t : k) {
        if (m.get(t) != 2) {
            answer *= t;
        }
    }
}

s 가 3일 경우, 두 개의 주사위 값만 같고 나머지는 다른 경우이다.
맵에 있는 key 값 중에 등장 횟수가 1인 값만 곱해서 계산했다.

else if (s == 4) {
    answer = 7;
            
    for (int i : n) {
        answer = Math.min(answer, i);
    }
}

s 가 4라면, 모든 주사위 값이 다른 경우이다.

Better Jeong