[Java] 백준 2665번 : 미로만들기

문제

n×n 바둑판 모양으로 총 n2개의 방이 있다. 일부분은 검은 방이고 나머지는 모두 흰 방이다. 검은 방은 사면이 벽으로 싸여 있어 들어갈 수 없다. 서로 붙어 있는 두 개의 흰 방 사이에는 문이 있어서 지나다닐 수 있다. 윗줄 맨 왼쪽 방은 시작방으로서 항상 흰 방이고, 아랫줄 맨 오른쪽 방은 끝방으로서 역시 흰 방이다.

시작방에서 출발하여 길을 찾아서 끝방으로 가는 것이 목적인데, 아래 그림의 경우에는 시작방에서 끝 방으로 갈 수가 없다. 부득이 검은 방 몇 개를 흰 방으로 바꾸어야 하는데 되도록 적은 수의 방의 색을 바꾸고 싶다.

아래 그림은 n=8인 경우의 한 예이다.

위 그림에서는 두 개의 검은 방(예를 들어 (4,4)의 방과 (7,8)의 방)을 흰 방으로 바꾸면, 시작방에서 끝방으로 갈 수 있지만, 어느 검은 방 하나만을 흰 방으로 바꾸어서는 불가능하다. 검은 방에서 흰 방으로 바꾸어야 할 최소의 수를 구하는 프로그램을 작성하시오.

단, 검은 방을 하나도 흰방으로 바꾸지 않아도 되는 경우는 0이 답이다.

입력

첫 줄에는 한 줄에 들어가는 방의 수 n(1 ≤ n ≤ 50)이 주어지고, 다음 n개의 줄의 각 줄마다 0과 1이 이루어진 길이가 n인 수열이 주어진다. 0은 검은 방, 1은 흰 방을 나타낸다.

출력

첫 줄에 흰 방으로 바꾸어야 할 최소의 검은 방의 수를 출력한다.

예제 입력 1

8
11100110
11010010
10011010
11101100
01000111
00110001
11011000
11000111

예제 출력 1

2

 

소스코드

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

public class Main {

	private static BufferedReader br;
	
	private static int N;
	private static int [][]graph, dp;
	private static int []dx= {-1,1,0,0};
	private static int []dy= {0,0,-1,1};
	
	static class Point implements Comparable<Point>{
		int r, c, cnt;

		public Point(int r, int c, int cnt) {
			super();
			this.r = r;
			this.c = c;
			this.cnt = cnt;
		}

		@Override
		public String toString() {
			return "Point [r=" + r + ", c=" + c + ", cnt=" + cnt + "]";
		}
		
		@Override
		public int compareTo(Point o) {	// 우선순위 큐에서 정렬할 기준을 정함
			return this.cnt-o.cnt;
		}
	}
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		br=new BufferedReader(new InputStreamReader(System.in));
		N=Integer.parseInt(br.readLine());
		graph=new int[N][N];
		dp=new int[N][N];
		
		for(int r=0;r<N;r++) {
			String s=br.readLine();
			for(int c=0;c<N;c++) {
				graph[r][c]=s.charAt(c)-'0';
				dp[r][c]=Integer.MAX_VALUE;
			}
		}
		
		int result=bfs();
		System.out.println(result);
	}

	private static int bfs() {	// bfs를 돌며 검은 방에서 흰 방으로 바꾸는 최소의 수가 나오면 개수를 리턴하도록 함
		PriorityQueue<Point> queue=new PriorityQueue<>();
		queue.add(new Point(0,0,0));
		dp[0][0]=0;
		
		while(!queue.isEmpty()) {
			Point p=queue.poll();
			
			if(p.r==N-1 && p.c==N-1) return p.cnt;	// 끝 방에 도착하면 결과값을 리턴
			for(int d=0;d<4;d++) {
				int nx=p.r+dx[d];
				int ny=p.c+dy[d];
				
				if(isIn(nx, ny)) {
					if(dp[nx][ny]<=dp[p.r][p.c]+1) continue;	// 현재 위치에 이미 dp값이 존재하며 그 값이 업데이트 될 값보다 작으면, 다음으로 넘어감
					if(graph[nx][ny]==1) {	// 현재 위치가 흰 방인 경우
						if(dp[nx][ny]>dp[p.r][p.c]) {
							dp[nx][ny]=dp[p.r][p.c];
							queue.add(new Point(nx, ny, dp[nx][ny]));
						}
					} else {	// 현재 위치가 검은 방인 경우
						if(dp[nx][ny]>dp[p.r][p.c]+1) {
							dp[nx][ny]=dp[p.r][p.c]+1;
							queue.add(new Point(nx, ny, dp[nx][ny]));
						}
					}
				}
			}
		}
		return 0;
	}
	
	private static boolean isIn(int r, int c) {
		return r>=0 && r<N && c>=0 && c<N;
	}
}

최소 개수를 구해야 하므로 bfs를 이용한 다익스트라를 적용해서 해결해주었습니다.