본문 바로가기

코딩 테스트/BAEKJOON

[Java] 2178 미로 탐색

문제

문제
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개의 정수로 미로가 주어진다. 
각각의 수들은 붙어서 입력으로 주어진다.

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

문제 풀이

DFS로 풀었다가 시간초과 이후 BFS로 풀었다.

로직 설명은 주석 확인.

더보기

DFS 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class N2178 {
    private static int N;
    private static int M;
    private static int result = Integer.MAX_VALUE;
    private static int cnt = 0;

    private static int[][] map;
    private static boolean[][] visited;

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

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");
        N = Integer.parseInt(input[0]);
        M = Integer.parseInt(input[1]);

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

        for(int i = 0; i < N; i++){
            String[] line = br.readLine().split("");
            for(int j = 0; j < M; j++){
                map[i][j] = Integer.parseInt(line[j]);
            }
        }

        dfs(0, 0);

        System.out.println(result);
    }

    public static void dfs(int x, int y){
        if(x == N - 1 && y == M - 1){
            result = Math.min(result, cnt + 1);
        }

        if(cnt > result)
            return;

        visited[x][y] = true;

        for(int i = 0; i < dr.length; i++){
            int nr = dr[i] + x;
            int nc = dc[i] + y;

            if(nr >= 0 && nc >= 0 && nr < N && nc < M && !visited[nr][nc] && map[nr][nc] == 1){
                cnt++;
                dfs(nr, nc);
                cnt--;
            }
        }

        visited[x][y] = false;
    }
}

자바 코드

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

public class N2178 {
    private static int N;
    private static int M;

    private static int[][] map;
    private static boolean[][] visited;

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

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");
        N = Integer.parseInt(input[0]);
        M = Integer.parseInt(input[1]);

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

        for(int i = 0; i < N; i++){
            String[] line = br.readLine().split("");
            for(int j = 0; j < M; j++){
                map[i][j] = Integer.parseInt(line[j]);
            }
        }

        bfs(0, 0);

        System.out.println(map[N - 1][M - 1]);
    }

    public static void bfs(int x, int y){
        // 큐에 x,y 좌표를 삽입
        Queue<int[]> q = new LinkedList<>();
        q.add(new int[] {x,y});

        // 큐가 빌때까지 실행
        while(!q.isEmpty()) {
            int[] now = q.poll();
            int nowX = now[0];
            int nowY = now[1];

            // 상하좌우의 좌표를 확인한다.
            for(int i = 0; i < 4; i++) {
                int nextX = nowX + dr[i];
                int nextY = nowY + dc[i];

                // map의 범위를 벗어나거나 방문을 한적이 있거나 이동할 수 없는 칸이면 continue;
                if (nextX < 0 || nextY < 0 || nextX >= N || nextY >= M)
                    continue;
                if (visited[nextX][nextY] || map[nextX][nextY] == 0)
                    continue;

                // 이동할 수 있는 다음 칸을 큐에 담는다.
                q.add(new int[] {nextX, nextY});
                // 다음칸에 현재 칸 + 1 의 값을 담고 다음 칸을 방문처리한다.
                map[nextX][nextY] = map[nowX][nowY] + 1;
                visited[nextX][nextY] = true;

                // 새로운 좌표가 담겼기 때문에 while 문은 종료되지 않고 다시 실행된다.
                // 모든 0이 아닌 좌표를 방문 처리 할 때까지 반복된다.
            }
        }
    }
}

 

'코딩 테스트 > BAEKJOON' 카테고리의 다른 글

[Java] 12996 Acka  (0) 2024.04.23
[Java] 1707 이분 그래프  (0) 2024.04.22
[Java] 2667 단지번호붙이기  (0) 2024.04.22
[Java] 1260 DFS와 BFS  (0) 2024.04.22
[Java] 1309 동물원  (0) 2024.04.21