[BOJ] 2503. 숫자 야구 (🥈, 브루트포스/구현)

🔗

풀이

가능한 모든 숫자를 구하고, 영수가 질문한 숫자들과 비교해 N 개의 질문을 모두 통과하는 숫자만이 정답이 될 가능성이 있는 숫자들이다.

3자리의 숫자는 1~9의 숫자들로 이뤄져있으며 중복되는 숫자는 존재할 수 없으므로 987 = 504개이다. permutation 함수를 통해 가능한 모든 숫자를 구한다.

영수가 질문한 숫자들과 비교하는 check 함수를 통해 모든 숫자들에 대해 질문한 N개의 숫자들과 비교한다. 스트라이크와 볼의 개수를 구하는데 이때 볼의 개수를 단순히 포함하고 있는지 여부로만 계산하면 스트라이크의 개수도 중복해서 구하기 때문에 구해진 볼의 개수에서 스트라이크의 개수를 제외해야 한다.

만약 도중에 스트라이크와 볼의 개수가 일치하지 않는다면 정답이 될 수 없으므로 0을 반환하고 그렇지 않다면 1을 반환한다.

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    static int N;
    static String[] questNumbers;
    static int[][] questStrikeBall;
    static String[] permuArr = new String[9 * 8 * 7];

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();

        questNumbers = new String[N];
        questStrikeBall = new int[N][2];
        for (int i = 0; i < N; i++) {
            questNumbers[i] = sc.next();
            questStrikeBall[i][0] = sc.nextInt();
            questStrikeBall[i][1] = sc.nextInt();
        }

        permutation();

        int answer = 0;
        for (String num : permuArr) {
            answer += check(num);
        }

        System.out.println(answer);
    }

    private static int check(String num) {
        for (int i = 0; i < N; i++) {
            int strike = 0;
            int ball = 0;
            String questNum = questNumbers[i]; // 영수가 질문한 숫자

            // 스트라이크 여부 비교
            for (int j = 0; j < 3; j++) {
                if (questNum.charAt(j) == num.charAt(j)) strike++;
            }

            // 볼 여부 비교
            for (int j = 0; j < 3; j++) {
                if (questNum.contains(String.valueOf(num.charAt(j)))) ball++;
            }

            ball -= strike;
            if (questStrikeBall[i][0] != strike || questStrikeBall[i][1] != ball) return 0;
        }

        return 1;
    }

    private static void permutation() {
        StringBuilder sb = new StringBuilder();
        int idx = 0;
        for (int i = 1; i < 10; i++) {
            if (sb.indexOf(String.valueOf(i)) != -1) continue;
            sb.append(i);
            for (int j = 1; j < 10; j++) {
                if (sb.indexOf(String.valueOf(j)) != -1) continue;
                sb.append(j);
                for (int k = 1; k < 10; k++) {
                    if (sb.indexOf(String.valueOf(k)) != -1) continue;
                    sb.append(k);
                    permuArr[idx] = sb.toString();
                    idx++;
                    sb.deleteCharAt(2);
                }
                sb.deleteCharAt(1);
            }
            sb.deleteCharAt(0);
        }
    }
}