Algorithm/BOJ
백준 9663번 C++
poopooreum
2023. 9. 3. 13:03
반응형
백준 9663번 N-Queen
https://www.acmicpc.net/problem/9663
9663번: N-Queen
N-Queen 문제는 크기가 N × N인 체스판 위에 퀸 N개를 서로 공격할 수 없게 놓는 문제이다. N이 주어졌을 때, 퀸을 놓는 방법의 수를 구하는 프로그램을 작성하시오.
www.acmicpc.net


정답 코드
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
using namespace std;
bool check[40];
bool check2[40];
bool check3[40];
int n,cnt1;
void func(int cur) {
if (cur == n) {
cnt1++;
return;
}
for (int i = 0; i < n; i++) {
if (check[i] || check2[i + cur] || check3[cur - i + n - 1]) //
continue;
check[i] = 1;
check2[i + cur] = 1;
check3[cur + -1 + n - i] = 1;
func(cur + 1);
check[i] = 0;
check2[i + cur] = 0;
check3[cur + -1 + n - i] = 0;
}
}
int main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> n;
func(0);
cout << cnt1;
}
문제 풀이
N-Queen문제는 N×N체스판 위에 queen을 서로 겹치지 않도록 놓는 문제입니다. 즉 동서남북 네 방향의 선이 겹치지 않아야 합니다. 다만 대각선 위치로 놓는 것은 상관없습니다. 그래서 저는 한 번 간곳은 더 이상 가지 않도록 하기 위해 백트래킹을 사용하였습니다.
사실 문제 자체는 N과M문제와 거의 유사합니다. 체크하기 위한 배열을 3개를 만들었고 방문할 때마다 1로 만들어주어서 더 이상 방문하지 않도록 하였습니다.
그리고 매개변수는 계속 cur로 돌려주면서 cur==n일때마다 cnt값을 ++해주었습니다
반응형