본문 바로가기

Algorithm/DFS

(C++) - 백준(BOJ) 9663번 : N-Queen 답

반응형

https://www.acmicpc.net/problem/9663

 

9663번: N-Queen

N-Queen 문제는 크기가 N × N인 체스판 위에 퀸 N개를 서로 공격할 수 없게 놓는 문제이다. N이 주어졌을 때, 퀸을 놓는 방법의 수를 구하는 프로그램을 작성하시오.

www.acmicpc.net

백트래킹 문제였습니다. 행을 기준으로 각 열에서 퀸이 놓일 수 있는 자리를 체크하고 dfs하였습니다.

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#include <iostream>
#include <cmath>
using namespace std;
int n;
int board[16];
int ans;
 
//행을 기준으로 열과 대각선을 확인
int check(int tuple) {
    for (int col = 0; col < tuple; col++)
    {
        if (board[tuple] == board[col] || abs(board[tuple] - board[col]) ==  tuple - col )
            return false;
    }
    return true;
}
 
void dfs(int tuple) {
    if (tuple == n)
        ans += 1;
    else
    {
        for (int col = 0; col < n; col++)
        {
            board[tuple] = col;
            if (check(tuple))
                dfs(tuple +1);
        }
    }
}
int main() {
    cin >> n;
    dfs(0);
    cout << ans << '\n';
}