[Java] 백준 2178번 미로 탐색

2023. 2. 22. 18:17나는 이렇게 학습한다/Algorithm

이 글은 백준 2178번 미로 탐색을 풀이한다. 코드는 Java로 구현하였다.

문제

N×M크기의 배열로 표현되는 미로가 있다.

1 0 1 1 1 1
1 0 1 0 1 0
1 0 1 0 1 1
1 1 1 0 1 1

미로에서 1은 이동할 수 있는 칸을 나타내고, 0은 이동할 수 없는 칸을 나타낸다. 이러한 미로가 주어졌을 때, (1, 1)에서 출발하여 (N, M)의 위치로 이동할 때 지나야 하는 최소의 칸 수를 구하는 프로그램을 작성하시오. 한 칸에서 다른 칸으로 이동할 때, 서로 인접한 칸으로만 이동할 수 있다.

위의 예에서는 15칸을 지나야 (N, M)의 위치로 이동할 수 있다. 칸을 셀 때에는 시작 위치와 도착 위치도 포함한다.

입력

첫째 줄에 두 정수 N, M(2 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 M개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.

예제 입력 1

4 6
101111
101010
101011
111011

예제 입력 2

4 6
110110
110110
111111
111101

예제 입력 3

2 25
1011101110111011101110111
1110111011101110111011101

예제 입력 4

7 7
1011111
1110001
1000001
1000001
1000001
1000001
1111111

출력

첫째 줄에 지나야 하는 최소의 칸 수를 출력한다. 항상 도착위치로 이동할 수 있는 경우만 입력으로 주어진다.

예제 출력 1

15

예제 출력 2

9

예제 출력 3

38

예제 출력 4

13

제한

  • 시간: 1초
  • 메모리: 192MB

풀이

접근

미로의 좌측 상단에서 우측 하단으로 가는 최단 경로의 길이를 찾는 문제이다.

일반적인 BFS 코드에서 방문을 했는지 체크하는 배열을 bool형으로 만드는데,

이번 문제에서는 int형으로 만들어서 시작점과의 거리를 계산할 수 있게 만들었다.

제일 먼저, 시작점으로부터의 거리를 저장하는 배열 dist를 하나 만들어서

Arrays.fill() 메서드를 이용해서 배열을 전부 -1로 초기화시켜둡니다.

그리고 BFS를 돌면서, 상하좌우의 칸을 볼 때 값을 잘 넣어주면 된다.

구현

코드

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

class Pair {

  int X, Y;

  Pair(int X, int Y) {
    this.X = X;
    this.Y = Y;
  }
}

public class baekjoon2178 {

  static int n, m;
  static int[][] board;
  static int[][] dist;
  static int[] dx, dy;
  static Queue<Pair> Q;

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

    board = new int[n][m];
    dist = new int[n][m];
    Q = new LinkedList<>();

    dx = new int[] { 1, 0, -1, 0 };
    dy = new int[] { 0, 1, 0, -1 };
    // 배열 값 세팅
    for (int i = 0; i < n; i++) {
      String s = br.readLine();
      for (int j = 0; j < m; j++) {
        board[i][j] = s.charAt(j) - '0';
      }
    }
    for (int d[] : dist) {
      Arrays.fill(d, -1);
    }

    // 시작
    Q.offer(new Pair(0, 0));
    dist[0][0] = 0;
    while (!Q.isEmpty()) {
      Pair cur = Q.poll();
      for (int dir = 0; dir < 4; dir++) {
        int nx = cur.X + dx[dir];
        int ny = cur.Y + dy[dir];
        if (nx < 0 || nx >= n || ny < 0 || ny >= m) continue;
        if (dist[nx][ny] >= 0 || board[nx][ny] != 1) continue;
        dist[nx][ny] = dist[cur.X][cur.Y] + 1;
        Q.offer(new Pair(nx, ny));
      }
    }

    System.out.println(dist[n - 1][m - 1] + 1);
  }
}

 

반응형