[프로그래머스] 37. 주사위 게임 3

입출력 예

입출력 예 설명

• 예제 1번에서 네 주사위 숫자가 모두 2로 같으므로 1111 × 2 = 2222점을 얻습니다. 따라서 2222를 return 합니다.

• 예제 2번에서 세 주사위에서 나온 숫자가 4로 같고 나머지 다른 주사위에서 나온 숫자가 1이므로 (10 × 4 + 1) = 41 = 1681점을 얻습니다. 따라서 1681을 return 합니다.
2
2

• 예제 3번에서 a, d는 6으로, b, c는 3으로 각각 같으므로 (6 + 3) × |6 - 3| = 9 × 3 = 27점을 얻습니다. 따라서 27을 return 합니다.

• 예제 4번에서 두 주사위에서 2가 나오고 나머지 다른 두 주사위에서 각각 5, 6이 나왔으므로 5 × 6 = 30점을 얻습니다. 따라서 30을 return 합니다.

• 예제 5번에서 네 주사위 숫자가 모두 다르고 나온 숫자 중 가장 작은 숫자가 2이므로 2점을 얻습니다. 따라서 2를 return 합니다.

코드 - map으로 푸는 법

import java.util.*;

class Solution {
    public int solution(int a, int b, int c, int d) {
        int[] dice = {a, b, c, d};
        Map<Integer, Integer> map = new HashMap<>();

        // 1. 주사위 숫자의 빈도를 저장
        for (int num : dice) {
            map.put(num, map.getOrDefault(num, 0) + 1); // 해시맵은 key갑 중복 안됨
        }

        // 2. 조건에 따른 점수 계산
        if (map.size() == 1) {  // (1) 네 개의 숫자가 같은 경우
            return 1111 * a;
        } 
        
        if (map.size() == 2) { // (2) 3개 같은 경우 or 2개씩 같은 경우
            List<Integer> keys = new ArrayList<>(map.keySet());
            int first = keys.get(0);
            int second = keys.get(1);
            int firstCount = map.get(first);
            int secondCount = map.get(second);

            if (firstCount == 3) { // 3개 같은 경우
                return (10 * first + second) * (10 * first + second);
            } else if (secondCount == 3) {
                return (10 * second + first) * (10 * second + first);
            } else { // 2개씩 같은 경우
                return (first + second) * Math.abs(first - second);
            }
        }
        
        if (map.size() == 3) { // (3) 2개 같은 경우 (q × r)
            int q = 0;
            int r = 0;
            for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
                if (entry.getValue() == 2) {
                } else if (q == 0) {
                    q = entry.getKey();
                } else {
                    r = entry.getKey();
                }
            }
            return q * r;
        }

        // (4) 모두 다른 경우
        return Collections.min(map.keySet());
    }
}

배열 정렬로 푸는 법

import java.util.*;
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];
        }
        else if (dice[0] == dice[2]) {  // 세 숫자가 같은 경우 1
            return (int) Math.pow((10 * dice[0] + dice[3]),2);
        }
        else if (dice[1] == dice[3]) {  // 세 숫자가 같은 경우 2
            return (int) Math.pow((10 * dice[1] + dice[0]),2);
        }
        else if (dice[0] == dice[1] && dice[2] == dice[3]) { // 2개씩 같은 값
            return (dice[0] + dice[2]) * Math.abs(dice[0] - dice[2]);
        }
        else if (dice[0] == dice[1] || dice[1] == dice[2] || dice[2] == dice[3]) { // 2개는 동일하고, 나머지 주사위가 다른 경우
            if (dice[0] == dice[1]) {
                return dice[2] * dice[3];
            }
            else if (dice[1] == dice[2]) {
                return dice[0] * dice[3];
            }
            else {
                return dice[0] * dice[1];
            }

        }
        else {
            return dice[0];
        }
    }
}