[Baekjoon Online Judge] 16956 - 늑대와 양 / Java

https://www.acmicpc.net/problem/16956

16956번: 늑대와 양

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {

	static int R, C;
	static char[][] map;
	static Queue<int[]> wQueue = new LinkedList<>();

	static int[] dr = { -1, 1, 0, 0 };
	static int[] dc = { 0, 0, -1, 1 };

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st = new StringTokenizer(br.readLine());

		R = Integer.parseInt(st.nextToken());
		C = Integer.parseInt(st.nextToken());
		map = new char[R][C];

		for (int i = 0; i < R; i++) {
			String str = br.readLine();
			for (int j = 0; j < C; j++) {
				map[i][j] = str.charAt(j);

				// 늑대인 경우 Queue에 추가
				if (map[i][j] == 'W') {
					wQueue.add(new int[] { i, j });
				}
			}
		}

		int[] curr;
		int r, c, nr, nc, result = 1;

		// 늑대의 사방을 울타리로 막기
		A: while (!wQueue.isEmpty()) {
			curr = wQueue.poll();
			r = curr[0];
			c = curr[1];

			for (int dir = 0; dir < 4; dir++) {
				nr = r + dr[dir];
				nc = c + dc[dir];

				// 경계 벗어나는 경우
				if (nr < 0 || nr >= R || nc < 0 || nc >= C) {
					continue;
				}

				// 바로 옆에 양이 있는 경우
				if (map[nr][nc] == 'S') {
					result = 0;
					break A;
				}
				// 빈칸인 경우
				else if (map[nr][nc] == '.') {
					map[nr][nc] = 'D';
				}
			}
		}

		// 결과 출력
		bw.write(result + "\n");
		if (result == 1) {
			for (int i = 0; i < R; i++) {
				for (int j = 0; j < C; j++) {
					bw.write(map[i][j]);
				}
				bw.write("\n");
			}
		}
		bw.flush();
	}
}