알고리즘

BOJ9663 N-Queen java

kkoon9 2022. 2. 11. 18:52

문제

N-Queen 문제는 크기가 N × N인 체스판 위에 퀸 N개를 서로 공격할 수 없게 놓는 문제이다.

N이 주어졌을 때, 퀸을 놓는 방법의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N이 주어진다. (1 ≤ N < 15)

출력

첫째 줄에 퀸 N개를 서로 공격할 수 없게 놓는 경우의 수를 출력한다.

import java.io.*;
import java.util.*;

class Main {
    static int N;
    static int answer = 0;

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        int[] chess = new int[N + 1];
        for (int i = 1; i <= N; i++) {
            chess[1] = i;
            BackTracking(chess, 1);
        }
        System.out.println(answer);
    }

    private static void BackTracking(int[] chess, int row) {
        if (row == N) {
            answer++;
            return;
        }
        for (int i = 1; i <= N; i++) {
            chess[row + 1] = i;
            if (isPossible(chess, row + 1)) {
                BackTracking(chess, row + 1);
            }
        }
    }

    private static boolean isPossible(int[] chess, int row) {
        for (int i = 1; i < row; i++) {
            if (chess[i] == chess[row]) return false;

            if (Math.abs(i - row) == Math.abs(chess[i] - chess[row])) return false;
        }
        return true;
    }

}