[백준 2503] 숫자 야구 (Java)

[백준 2503] 숫자 야구 (Java)

출처 : 링크

완탐 문제다.

처음에 규칙을 어떻게 가져와야할까 고민했지만, 그냥 다 돌면서 해당하는 규칙과 비교하고 성립하는 수만 고르면 된다.

123~987까지 for문으로 돌아도 되지만, 중복되는 수를 없애기위해 순열을 만들어 풀었다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class Main_2503_숫자야구 {
 
    static int N;
    static String[] str;
    static int[] strike;
    static int[] ball;
    
    static int[] set;
    static int[] num = {1,2,3,4,5,6,7,8,9}; // 숫자 1~9 저장
    static int cnt;
    
    public static void main(String[] args) throws Exception {
 
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        N = Integer.parseInt(br.readLine().trim());
        str = new String[N];
        strike = new int[N];
        ball = new int[N];
        
        for (int i = 0; i < N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        
            str[i] = st.nextToken();
            strike[i] = Integer.parseInt(st.nextToken());
            ball[i] = Integer.parseInt(st.nextToken());
        }
    
        
        set = new int[9];
        
        for (int i = 0; i < 9; i++) {
            set[i] = i;
        }
        
        cnt = 0;
        
        perm(set, 0, 9, 3); // 순열 돌리기
        
        System.out.println(cnt);
    }
    
    public static void perm(int[] set, int size, int N, int K) {
        
        if(size == K){
            String baseball = "";
            for (int i = 0; i < size; i++) {
                baseball += Integer.toString(num[set[i]]);
            }
            if(checking(baseball)) cnt++;
            return;
        }
        
        
        for (int i = size; i < N; i++) {
            swap(set, i, size);
            perm(set, size+1, N, K);
            swap(set, i, size);
        }
    }
    
    public static boolean checking(String baseball) {
        String[] chk = baseball.split("");
        
        for (int i = 0; i < N; i++) {
            
            String[] chk2 = str[i].split("");
            int s = 0; // 스트
            int b = 0; // 볼
            
            for (int j = 0; j < chk.length; j++) { // 스트라이크 체크
                if(chk[j].equals(chk2[j]))
                    s++;
            }
            
            for (int j = 0; j < chk.length; j++) { // 볼 체크
                for (int z = 0; z < chk2.length; z++) {
                    if(j==z) continue;
                    else {
                        if(chk[j].equals(chk2[z]))
                            b++;
                    }
                }
            }
            
            if(strike[i] != s || ball[i] != b) // 틀린게 있으면 해당x
                return false;
            
        }
        
        return true;
    }
    
    public static void swap(int[] set, int i, int index){
        int temp = set[i];
        set[i] = set[index];
        set[index] = temp;
    }
 
}