[SWEA]N-Queen

백트래킹, DFS

Posted by kyoungIn on March 19, 2019

N-Queen

링크

풀이

옛날에 시간초과로 못풀었던 문제인데 풀었다!!

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
38
39
40
41
42
43
44
45
46
#include <iostream>
#include <math.h>
using namespace std;
int arr[11][11];
int n,ans=0;
void dfs(int y,int x){
    if(y==n-1){
        ++ans;
        return;
    }
    arr[y][x]=1;
    for(int i=y+1;i<n;i++){
        arr[i][x]-=1;
        for(int j=0;j<n;j++){
            if(abs(i-y)==abs(j-x))
                arr[i][j]-=1;
        }
    }
    for(int i=0;i<n;i++){
        if(arr[y+1][i] == 0)
            dfs(y+1,i);
    }
    
    for(int i=y+1;i<n;i++){
        arr[i][x]+=1;
        for(int j=0;j<n;j++){
            if(abs(i-y)==abs(j-x))
                arr[i][j]+=1;
        }
    }
    arr[y][x]=0;
}
int main(){
    int t; cin >> t;
    for(int T=1;T<=t;T++){
        cin >> n;
        ans=0;
        for(int i=0;i<n;i++)
            dfs(0,i);
        
        
        cout << "#"<<T<< " "<<ans <<'\n';
        
    }
}