[백준]#6987 월드컵

문제

월드컵 조별 최종 예선에서는 6개국으로 구성된 각 조별로 동일한 조에 소속된 국가들과 한 번씩, 각 국가별로 총 5번의 경기를 치른다. 조별리그가 끝난 후, 기자가 보내온 각 나라의 승, 무승부, 패의 수가 가능한 결과인지를 판별하려고 한다. 다음은 가능한 결과와 가능하지 않은 결과의 예이다.

네 가지의 결과가 주어질 때 각각의 결과에 대하여 가능하면 1, 불가능하면 0을 출력하는 프로그램을 작성하시오.

입력

첫째 줄부터 넷째 줄까지 각 줄마다 6개국의 결과가 나라별로 승, 무승부, 패의 순서로 빈칸을 하나 사이에 두고 18개의 숫자로 주어진다. 승, 무, 패의 수는 6보다 작거나 같은 자연수 또는 0이다.

출력

입력에서 주어진 네 가지 결과에 대하여 가능한 결과는 1, 불가능한 결과는 0을 빈칸을 하나 사이에 두고 출력한다.

예제 입력 1

5 0 0 3 0 2 2 0 3 0 0 5 4 0 1 1 0 4
4 1 0 3 0 2 4 1 0 1 1 3 0 0 5 1 1 3
5 0 0 4 0 1 2 2 1 2 0 3 1 0 4 0 0 5
5 0 0 3 1 1 2 1 2 2 0 3 0 0 5 1 0 4

예제 출력 1

1 1 0 0

풀이

이 문제는 재귀를 이용해서 풀 수 있는 문제였다.

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {
    static int[][] result;
    static int[] ans;
    static int t;

    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        ans = new int[4];

        for(t=0; t<4; t++) {
            String[] input = br.readLine().split(" ");
            result = new int[6][3];
            int sum = 0;

            for(int i=0; i<18; i++) {
                int x = Integer.parseInt(input[i]);
                result[i/3][i%3] = x;
                sum += x;
            }

            if(sum!=30)
                continue;

            dfs(0, 1);
        }

        for(int i=0; i<4; i++)
            System.out.print(ans[i]+" ");
    }

    public static void dfs(int a, int b) {
        if(a==6) {
            ans[t] = 1;
        }

        else if(b==6)
            dfs(a+1, a+2);

        else {
            for (int i=0; i<3; i++) {
                if (result[a][i]>=1 && result[b][2-i]>=1) {
                    result[a][i]--;
                    result[b][2-i]--;
                    dfs(a, b+1);
                    result[a][i]++;
                    result[b][2-i]++;
                }
            }
        }
    }
}