[Java] 백준 14940 쉬운 최단거리

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

난이도

실버1

문제

지도가 주어지면 모든 지점에 대해서 목표지점까지의 거리를 구하여라.
문제를 쉽게 만들기 위해 오직 가로와 세로로만 움직일 수 있다고 하자.

입력

지도의 크기 n과 m이 주어진다. n은 세로의 크기, m은 가로의 크기다.(2 ≤ n ≤ 1000, 2 ≤ m ≤ 1000)
다음 n개의 줄에 m개의 숫자가 주어진다. 0은 갈 수 없는 땅이고 1은 갈 수 있는 땅, 2는 목표지점이다. 입력에서 2는 단 한개이다.

출력

각 지점에서 목표지점까지의 거리를 출력한다. 원래 갈 수 없는 땅인 위치는 0을 출력하고, 원래 갈 수 있는 땅인 부분 중에서 도달할 수 없는 위치는 -1을 출력한다.

내 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());

		int[] goal = new int[3];

		int[][] map = new int[N][M];
		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine());
			for (int j = 0; j < M; j++) {
				map[i][j] = Integer.parseInt(st.nextToken());
				if (map[i][j] == 2) {
					goal[0] = i;
					goal[1] = j;
				}
			}
		}

		boolean[][] visited = new boolean[N][M];
		int[][] ans = new int[N][M];

		int[] dx = { -1, 0, 1, 0 };
		int[] dy = { 0, 1, 0, -1 };

		Queue<int[]> q = new LinkedList<>();
		q.add(goal);
		visited[goal[0]][goal[1]] = true;

		while (!q.isEmpty()) {
			int[] now = q.poll();
			for (int i = 0; i < 4; i++) {
				int nx = now[0] + dx[i];
				int ny = now[1] + dy[i];
				if (0 > nx || nx >= N || 0 > ny || ny >= M) {
					continue;
				}
				if (visited[nx][ny]) {
					continue;
				}
				if (map[nx][ny] == 0) {
					continue;
				}
				q.add(new int[] { nx, ny, now[2] + 1 });
				visited[nx][ny] = true;
				ans[nx][ny] = now[2] + 1;
			}
		}

		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				if (map[i][j] == 1 && !visited[i][j]) {
					sb.append(-1).append(" ");
				} else {
					sb.append(ans[i][j]).append(" ");
				}
			}
			sb.append("\n");
		}

		System.out.print(sb);
	}
}

간단한 BFS 구현 문제였다.
하지만 갈 수 없는 곳을 -1로 표시해야하는데 이 부분 깜빡해서 한번 틀렸다.

그리고 출력을 할 때 그냥 System.out.print(ans[i][j] + " "); 이런식으로 했었는데 시간이 너무 오래 걸리길래 sb.append(ans[i][j]).append(" ");이렇게 StringBuilder에 담아서 한번에 출력하니 시간이 반이 줄었다.

데이터의 크기가 작으면 두 방식의 차이가 크지 않지만, 대량의 출력을 할 경우에는 StringBuilder에 담아서 출력하는 것이 훨씬 빠르고 효율적이다.