Untitled

 avatar
unknown
plain_text
a year ago
1.5 kB
5
Indexable
import java.util.Scanner;

public class ColorCombinationBacktracking {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int T = sc.nextInt();
        for (int t = 1; t <= T; t++) {
            int n = sc.nextInt();
            int[][] matrix = new int[n][n];
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    matrix[i][j] = sc.nextInt();
                }
            }

            int count = 0;
            int[] color = new int[n * n];
            backtrack(matrix, color, 0);
            System.out.println("Case #" + t);
            System.out.println(count);
        }
    }

    private static void backtrack(int[][] matrix, int[] color, int i) {
        if (i == matrix.length * matrix.length) {
            count++;
            return;
        }

        for (int c = 0; c < 4; c++) {
            if (isSafe(matrix, color, i, c)) {
                color[i] = c;
                backtrack(matrix, color, i + 1);
            }
        }
    }

    private static boolean isSafe(int[][] matrix, int[] color, int i, int c) {
        int row = i / matrix.length;
        int col = i % matrix.length;

        for (int j = 0; j < matrix.length; j++) {
            if (matrix[row][j] == 1 && color[j * matrix.length + c] == c) {
                return false;
            }
            if (matrix[j][col] == 1 && color[i + j * matrix.length] == c) {
                return false;
            }
        }

        return true;
    }
}
Editor is loading...