Algorithm

[BOJ] 9663. N-Queen

프로그래민 2020. 3. 27. 00:39
반응형

백트래킹의 전형적인 문제인 N-Queen 문제이다.

체스판에 서로 공격할 수 없도록 N개의 Queen을 배치하는 문제인데, Queen의 움직임 특성상 한 행에 한 개의 Queen 만을 가질수 있게 됨으로 한행을 depth로 보고 dfs를 실행하였다. 보통의 백트래킹은 visit함수로만 방문 여부를 결정하였는데 이 문제에선 대각선이랑 같은 열에 Queen이 존재해야하는지 봐야함으로 check함수를 이용하여 방문 가능여부를 한번더 확인해주는 과정을 거쳐 문제를 해결하였다. 

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
47
48
49
50
51
52
53
54
package chap07;
 
 
public class Main_bj_9663_NQueen {
    
    static boolean[][] visit;
    static int N;
    static int count;
    
    public static void main(String[] args) throws Exception {
        BufferedReader br= new BufferedReader(new InputStreamReader(System.in));
        N=Integer.parseInt(br.readLine());
        
        visit=new boolean[N][N];
        count=0;
        
        dfs(0);
        System.out.println(count);
        
    }
    
    static void dfs(int depth) {
        if(depth==N) {
            count++;
            return;
        }
        for(int col=0;col<N;col++) {
            if(visit[depth][col]==false && check(depth,col)==true) {
                visit[depth][col]=true;
                dfs(depth+1);
                visit[depth][col]=false;
            }
        }
    }
    
    static boolean check(int row,int col) {
        for(int len=1;len<=row;len++) {
            if(visit[row-len][col]==true) {
                return false;
            }
            if(col-len >= 0 && visit[row-len][col-len]==true) {
                return false;
            }
            if(col+len <&& visit[row-len][col+len]==true) {
                return false;
            }
        }
        return true;
    }
    
}
 
http://colorscripter.com/info#e" target="_blank" style="color:#4f4f4ftext-decoration:none">Colored by Color Scripter
                                                     
반응형

'Algorithm' 카테고리의 다른 글

[BOJ] 1182. 부분수열의 합  (0) 2020.03.30
[BOJ] 1707. 이분 그래프  (0) 2020.03.29
[SWEA] 5656. 벽돌 깨기  (0) 2020.03.24
[SWEA] 4012. 요리사  (0) 2020.03.24
[SWEA] 1868. 파핑파핑 지뢰찾기  (0) 2020.03.23