[Programmers 코딩 기초 트레이닝 : Java] 주사위 게임 3

🎲 주사위 게임 3 (프로그래머스)

링크 : https://school.programmers.co.kr/learn/courses/30/lessons/181916

프로그래머스

문제 요약

• 4개의 주사위 숫자(a, b, c, d)가 주어짐.
• 규칙에 따라 점수 계산:
  1. 네 주사위 모두 같음 → 1111 × 주사위 숫자
  2. 네 주사위 모두 다름 → 최소 숫자
  3. 세 주사위 같고 하나 다름 → (10 × 동일 숫자 + 다른 숫자)^2
  4. 두 주사위 같고 나머지 두 주사위 각각 다름 → (같은 주사위 숫자 + 다른 주사위 숫자) × |같은 주사위 숫자 - 다른 주사위 숫자|
  5. 두 쌍 같은 경우 → (첫 쌍 숫자 + 두 번째 쌍 숫자) × |첫 쌍 숫자 - 두 번째 쌍 숫자|

---

1️⃣ 처음 작성한 코드

 

class Solution {
    public int solution(int a, int b, int c, int d) {
        int answer = 0; // 블록 밖에서 선언

        // 1. 네 주사위 모두 같음
        if(a == b && b == c && c == d) {
            answer = 1111 * a;

        // 2. 네 주사위 모두 다름
        } else if(a!=b && a!=c && a!=d &&
                  b!=c && b!=d &&
                  c!=d) {
            answer = Math.min(Math.min(a,b), Math.min(c,d));

        // 3. 세 주사위가 같고 나머지 하나 다름
        } else if((a==b && b==c) || (a==b && b==d) || 
                  (a==c && c==d) || (b==c && c==d)) {
            
            if(a==b && b==c) answer = (int)Math.pow(10*a + d, 2);
            else if(a==b && b==d) answer = (int)Math.pow(10*a + c, 2);
            else if(a==c && c==d) answer = (int)Math.pow(10*a + b, 2);
            else if(b==c && c==d) answer = (int)Math.pow(10*b + a, 2);

        // 4. 두 주사위가 같고 나머지 두 주사위가 서로 다르거나 두 쌍 같은 경우
        } else {
            // 두 쌍 같은 경우
            if(a==b && c==d) answer = (a + c) * Math.abs(a - c);
            else if(a==c && b==d) answer = (a + b) * Math.abs(a - b);
            else if(a==d && b==c) answer = (a + b) * Math.abs(a - b);
            // 하나만 중복
            else if(a==b) answer = c * d;
            else if(a==c) answer = b * d;
            else if(a==d) answer = b * c;
            else if(b==c) answer = a * d;
            else if(b==d) answer = a * c;
            else if(c==d) answer = a * b;
        }

        return answer;
    }
}

장점

• 모든 경우를 조건문으로 명시 → 논리 흐름 직관적
• 간단한 if-else 구조로 바로 이해 가능

단점

• 조건문이 길고 중복됨 → 유지보수 어려움
• 숫자가 많아질수록 조건이 폭발적으로 증가
• 연산 및 가독성 측면에서 비효율적

---

2️⃣ 개선한 코드 (Stream + 배열 기반)

 

import java.util.*;

class Solution {
    private int findSingleIndex(int a, int b, int c, int d) {
        if(a==b && b==c) return 3;
        else if(a==b && b==d) return 2;
        else if(a==c && c==d) return 1;
        else return 0;
        
    }
    
    public int solution(int a, int b, int c, int d) {
        int answer = 0; 
        
        // 배열로 생성
        int[] nums = {a, b, c, d};
        int[] distinctNums = Arrays.stream(nums)
                    .distinct() // 중복 제거
                    .sorted() // 정렬
                    .toArray();
        
        // System.out.println(Arrays.toString(nums)); // 배열 출력
        // 1. 모두 같은 숫자 => 배열 사이즈 1
        if(distinctNums.length == 1) {
            answer = 1111*distinctNums[0];
        
        // 2. 모두 다른 숫자 => 배열 사이즈 4
        } else if(distinctNums.length == 4) {
            answer = Math.min(Math.min(distinctNums[0], distinctNums[1]), Math.min(distinctNums[2], distinctNums[3]));        
        // 3. 세 주사위가 같은 숫자 => 배열 사이즈 2
        } else if (distinctNums.length == 2) {
            
            int cnt1 = 0;
            int cnt2 = 0;
            // 몇개가 같은지 판별
            for(int n: nums) {
                if(n==distinctNums[0]) cnt1++;
                else cnt2++;
            }
            
            // 3개 같고, 1개 다름            
            if(cnt1==3 || cnt2==3) {
                int p=0;
                int q=0;
                // 어떤 숫자가 3개씩 나왔는지 판별
                switch(findSingleIndex(a, b, c, d)) {
                    case 0: p=b; q=a; break;
                    case 1: p=a; q=b; break;
                    case 2: p=a; q=c; break;
                    case 3: p=a; q=d; break;
                }
                answer = (10*p + q) * (10 * p + q);
            } else {
                int p = distinctNums[0];
                int q = distinctNums[1];
                answer = (p+q) * Math.abs(p-q);
            }
        // 4. 두 주사위가 같고, 2개가 각각 다름 => 
        } else {
            // 뭐가 2번 나온지 확인 후 다른것 합하기
            int cnt1 = 0;
            int cnt2 = 0;
            int cnt3 = 0;
            
            int q = 0;
            int r = 0;
            
            for(int n: nums) {
                if(n == distinctNums[0]) cnt1++;
                else if(n == distinctNums[1]) cnt2++;
                else cnt3++;
           }

            if(cnt1 == 2) {
                q = distinctNums[1];
                r = distinctNums[2]; 
            } else if (cnt2 == 2) {
                q = distinctNums[0];
                r = distinctNums[2]; 
            } else {
                q = distinctNums[0];
                r = distinctNums[1]; 
            }
            answer = q*r;
        }

        return answer;
    }
}

장점

• 배열 + Stream API 활용 → 숫자 중복 제거, 정렬, 최소값 확인 간편
• 조건문이 단순화 → 유지보수 용이
• 연산량 최소화 (필요한 경우에만 계산)

단점

• 처음 보는 사람에게는 Stream API와 distinct() 등 익숙하지 않을 수 있음
• switch문과 배열 인덱스 연산이 조금 복잡

---

3️⃣ 두 코드 비교

구분처음 코드개선 코드

가독성	조건문 길고 중복 많음	배열 + Stream, 경우 나누기 간결
유지보수	어려움, 조건 추가 시 복잡	용이, 숫자가 늘어나도 구조 유지 가능
성능	상수 개수라 큰 문제 없음, 하지만 조건문 많음	Stream과 distinct 사용 → O(n) 내외, 작은 상수라 체감 없음
복잡도	O(1) (항상 4개 숫자)	O(1), Stream 사용으로 약간 오버헤드 있지만 상수 수준

---

4️⃣ 최종 평가

• 처음 코드: 직관적이지만, 조건문이 길고 유지보수가 어렵다.
• 개선 코드: 가독성, 유지보수, 구조적 명확성에서 우수.
  • Stream + 배열 접근으로 중복 제거, 정렬, 최소값 탐색이 간단해짐.
  • 실제 성능 차이는 미미하지만, 코드 확장성과 가독성 측면에서 훨씬 좋음.

💡 결론: 현업 코드나 협업 환경에서는 개선 코드를 추천