[JAVA] 백준 (실버1) 14940번 쉬운 최단거리

AIR·2024년 3월 22일

그래프 탐색 백준 실버 코딩테스트

코딩 테스트 문제 풀이

목록 보기

63/135

링크

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

문제 설명

정답률 37.136%
지도가 주어지면 모든 지점에 대해서 목표지점까지의 거리를 구하여라.

문제를 쉽게 만들기 위해 오직 가로와 세로로만 움직일 수 있다고 하자.

입력 예제

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

15 15
2 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1 0 0 0 0 1
1 1 1 1 1 1 1 1 1 1 0 1 1 1 1
1 1 1 1 1 1 1 1 1 1 0 1 0 0 0
1 1 1 1 1 1 1 1 1 1 0 1 1 1 1

출력 예제

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

풀이

최소 이동거리를 거리를 구하는 문제이므로 BFS를 사용한다. 목적지에 대한 최소 이동거리를 구해야하므로 우선 목적지의 좌표부터 탐색한 뒤 BFS를 진행한다. 갈 수 있는 곳이나 0으로 둘러싸여 탐색을 못하면 원소를 -1으로 할당한다.

코드

//백준
public class Main {

    private static final int[] dr = {-1, 0, 1, 0};
    private static final int[] dc = {0, 1, 0, -1};
    private static int[][] map;
    private static boolean[][] visit;
    private static int M;
    private static int N;

    public static void main(String[] args) throws IOException {

        System.setIn(new FileInputStream("src/input.txt"));

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        N = Integer.parseInt(st.nextToken());  //세로
        M = Integer.parseInt(st.nextToken());  //가로

        map = new int[N][M];
        visit = new boolean[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());
            }
        }

        //목적지 좌표
        int[] destination = new int[2];

        //목적지 탐색
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if (map[i][j] == 2) {
                    destination[0] = i;
                    destination[1] = j;
                }
            }
        }

        bfs(destination[0], destination[1]);

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                //도달하지 않은 곳 -1
                if (!visit[i][j] && map[i][j] == 1) {
                    map[i][j] = -1;
                }
                sb.append(map[i][j]).append(" ");
            }
            if (i != N - 1) {
                sb.append("\n");
            }
        }

        System.out.println(sb);
    }

    private static void bfs(int r, int c) {

        map[r][c] = 0;
        Queue<int[]> queue = new LinkedList<>();
        queue.add(new int[]{r, c});

        while (!queue.isEmpty()) {
            int[] curPos = queue.poll();
            int curR = curPos[0];
            int curC = curPos[1];

            for (int i = 0; i < 4; i++) {
                int nextR = curR + dr[i];
                int nextC = curC + dc[i];

                if (nextR < 0 || nextC < 0 || nextR >= N || nextC >= M) {
                    continue;
                }

                if (map[nextR][nextC] == 1 && !visit[nextR][nextC]) {
                    visit[nextR][nextC] = true;
                    queue.add(new int[]{nextR, nextC});
                    map[nextR][nextC] = map[curR][curC] + 1;
                }
            }

        }
    }

}