[프로그래머스] N-Queen / C++

문제 설명

가로, 세로 길이가 n인 정사각형으로 된 체스판이 있습니다. 체스판 위의 n개의 퀸이 서로를 공격할 수 없도록 배치하고 싶습니다.

예를 들어서 n이 4인경우 다음과 같이 퀸을 배치하면 n개의 퀸은 서로를 한 번에 공격할 수 없습니다.

체스판의 가로 세로의 세로의 길이 n이 매개변수로 주어질 때, n개의 퀸이 조건에 만족 하도록 배치할 수 있는 방법의 수를 return 하는 solution함수를 완성해주세요.

 

제한 조건

• 퀸(Queen)은 가로, 세로, 대각선으로 이동할 수 있습니다.
• n은 12이하의 자연수입니다.

 

입출력 예

n	result
4	2

 

 

코드

#include <string>
#include <vector>

using namespace std;

vector<int> board;
int answer;

void dfs(int r, int n)
{
    int i, j;
    
    // 마지막 퀸까지 놓은 경우
    if (r > n)
    {
        answer++;
        return;
    }
    // r행의 모든 열에 대해 퀸을 놓을 수 있는지 확인
    for (i = 1; i <= n; i++)
    {
        board[r] = i;
        for (j = 1; j < r; j++)
        {
       	    // 같은 열에 있거나 대각선에 위치하는 경우 퀸을 놓을 수 없음
            if (board[j] == i || abs(r - j) == abs(i - board[j]))
            {
                board[r] = 0;
                break;
            }
        }
        // 퀸을 놓았다면 다음 행에 대하여 dfs를 수행
        if (board[r])
            dfs(r + 1, n);
    }
}

int solution(int n)
{
    board.resize(n + 1, 0);
    dfs(1, n);
    return answer;
}

 

링크

코딩테스트 연습 - N-Queen

inbdni/Programmers