[백트래킹] C11 백준 9663 N-Queen 풀이

문제

• 9663

풀이 과정

• Queen은 상하좌우대각선으로 공격할 수 있음
  • 행, 열, \대각선, /대각선에 1개의 퀸만을 놓을 수 있음
  • 베이스코드는 N*N 체스판 이기 때문에 행이 N에 도달하면, 즉 N에 퀸을 놓을 수 있다면 카운팅
  • 행, 열, 왼, 오대각선에 무조건 1개만 배치되어야 하기 때문에 3가지의 경우의 수를 확인해야 함
• 문제는 대각선인데...
  • \ 대각선일 경우 x-y를 확인하면 됨
    • 근데 음수로 가는 것을 방지하기 위해 n을 더해줌
  • / 대각선일 경우 x+y를 확인하면 됨
• 그래서 열을 확인하는 col 1개, 각각의 대각선을 확인하는 left, right 2개 해가지고 총 3개로 구현하고 백트래킹

#include <stdio.h>

#define MAX 15

int n;
int result;
int col[MAX];
int left[2*MAX];
int right[2*MAX];

void func(int row) {
    if (row == n) {
        result++;
        return;
    }
    
    for (int i=0; i<n; i++) {
        if (col[i] == 0 &&
           left[row - i + n] == 0 &&
           right[row + i] == 0) {
            col[i] = 1;
            left[row - i + n] = 1;
            right[row + i] = 1;
            
            func(row+1);
            
            col[i] = 0;
            left[row - i + n] = 0;
            right[row + i] = 0;
        }
    }
}

int main() {
    scanf("%d", &n);
    func(0);
    printf("%d", result);
    return 0;
}