알고리즘

[백준] | JAVA, 자바 | 14940번 - 쉬운 최단거리

inttype 2025. 3. 4. 22:01

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


문제 요약 :

1. 지도의 크기 n과 m 입력받기

2. n x m 크기의 지도 정보 입력받기

3. 값이 2인 지점에서 시작

4. 2인 지점을 기준으로 상하좌우 이동해 목표지점까지의 거리를 계산

5. 계산된 지도 정보 출력


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

public class Main {
    static class Coordinate {
        int x;
        int y;
        public Coordinate(int x, int y) {
            this.x = x;
            this.y = y;
        }
    }
    private static int N, M;
    private static int[][] map, distance;
    private static boolean[][] visited;
    private static Coordinate target;
    private static Queue<Coordinate> queue;
    private static int[][] delta = {{0,1},{1,0},{-1,0},{0,-1}};
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        // 1. 지도의 크기 n과 m 입력받기
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        map = new int[N][M];
        // 목표 지점까지의 거리를 저장할 2차원 배열
        distance = new int[N][M];
        visited = new boolean[N][M];

        // 2. n x m 크기의 지도 정보 입력받기
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < M; j++) {
                int n = Integer.parseInt(st.nextToken());
                // 시작점 저장
                if (n == 2) {
                    target = new Coordinate(i, j);
                    visited[i][j] = true;
                    n = 0;
                }
                map[i][j] = n;
                // 초기값 -1로 설정
                distance[i][j] = -1;
            }
        }

        queue = new ArrayDeque<>();
        queue.add(target);

        move();
        // 5. 계산된 지도 정보 출력
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if (map[i][j] == 0) {
                    System.out.print(0 + " ");
                } else {
                    System.out.print(distance[i][j] + " ");
                }
            }
            System.out.println();
        }
    }

    private static void move() {

        while (!queue.isEmpty()) {
            Coordinate current = queue.poll();
            int x = current.x;
            int y = current.y;

            // 4. 상화좌우 이동해 목표지점까지의 거리를 계산
            for (int i = 0; i < 4; i++) {
                int nr = x + delta[i][0];
                int nc = y + delta[i][1];

                if (nr >= 0 && nr < N && nc >= 0 && nc < M) {
                    if (map[nr][nc] == 1 && distance[nr][nc] == -1) {
                        distance[nr][nc] = distance[x][y] + 1;
                        queue.offer(new Coordinate(nr, nc));
                    }
                }
            }
        }

    }
}