[백준] 9663. N-Queen(골드5)

ERror.ASER·2021년 2월 23일
0

백준

목록 보기
19/69
post-thumbnail

백준(골드5) - 9663. N-Queen(골드5)



풀이

체스 룰을 모르는데 알고리즘으로 풀려고 하니까 너무 피곤했다..
문제는 안어려운데 퀸의 이동이 이게 맞나? 하고 확신을 못했다.
결국 인터넷에 퀸의 이동방향을 쳐보고 나서야 문제를 풀었다;ㅡ;

해당 구역이 비어있고, 다른 퀸들에게 공격받지 않은지만 체크(isValid) 해주면 되었다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
	static int n,count;
	static int[][] map;
	static boolean[][] isSelected;
	//대각선 방향
	static int[] x = {-1,-1,1,1};
	static int[] y = {-1,1,-1,1};
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		n = Integer.parseInt(br.readLine());
		map = new int[15][15];
		count = 0;

		dfs(0);
		System.out.println(count);
	}

	public static boolean isValid(int i,int cnt) {
		for(int j = 0; j < cnt; j++) {
			if(map[i][j]==1) return false;//그 행에 퀸이 있으면 안된다.
		}

		for(int j=1; j<=n; j++) {//대각선 체크
			for(int k=0; k<4; k++) {
				int dx = i+j*x[k];
				int dy = cnt + j*y[k];
				//System.out.println("i: "+i+" cnt: "+cnt+" dx:"+dx+" dy:"+dy);
				if(dx>=0 && dy>=0 && dx<n && dy<n) {
					if(map[dx][dy]==1) return false;
				}
			}
		}
		

		return true;
	}

	public static void dfs(int cnt) {
		if(cnt == n) {
			count++;
			return;
		}

		for(int i=0; i<n; i++) {
			if(map[i][cnt] == 0 && isValid(i,cnt)) {//비어있고 놓을 수 있나 체크
				map[i][cnt] = 1;
				dfs(cnt+1);
				map[i][cnt] = 0;
			}
		}

	}

}
profile
지우의 블로그

0개의 댓글