본문 바로가기

Algorithm

[백트래킹 단계] 백준 9663번 N-Queen

반응형

문제

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

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

입력

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

출력

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

 

진짜 바보다...

NxN에서 N개 퀸을 넣으려면 모든 열에 1개씩 들어가야 한다.

퀸은 모든 열 행 대각선을 다 차지하므로, 같은 열에 있을 수 없기 때문..

따라서 0열 부터 시작해서 모든 행에대해 조사해주면 되므로 i,j 좌표는 필요 없...

 

 

#include <iostream>
#include <vector>
#include<queue>
#include <string.h>
#include <map>
#include<algorithm>
using namespace std;
int N;
vector<int> vec;
int cnt = 0;
int col[15];

bool ok(int idx)
{
	for (int i = 0; i < idx; i++)
	{
		if (col[i] == col[idx] || abs(col[i] - col[idx]) == idx - i)
		{
			return false;
		}
	}

	return true;
}

void DFS(int cur)
{
	if (cur == N)
	{
		cnt++;
		return;
	}

	for (int i = 0; i < N; i++)
	{
		col[cur] = i;
		if (ok(cur))
		{
			DFS(cur + 1);
		}
	}
	
}


int main() {
	cin >> N;
	for (int i = 0; i < N; i++)
	{
		col[0] = i;
		DFS(1);
	}
	cout << cnt << endl;

}
반응형