1261번: 알고스팟 [JAVA]

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

1261번: 알고스팟

 

😂 N, M을 입력받고, 그대로 map = new int[N][M] 으로 설정해서 틀렸었다.

map = new int[M][N]으로 설정해야 된다....

 

접근 방식) 다익스트라

1. int[][] map을 생성해서 미로를 저장한다.

    부숴야하는 벽의 최소 개수를 저장할 int[][] result 배열을 Integer.MAX_VALUE로 초기화한다.

 

2. map[0][0]을 시작점으로 BFS 시작한다.

    각 지점마다 상하좌우를 돌면서 부숴야하는 벽의 개수를 구해 result[][] 값을 갱신한다.

 

3. BFS에서는 queue 대신, 부숴야하는 벽의 개수가 적은 순서로 먼저 살펴보기 위해 우선순위 큐를 사용한다.

  ⭐우선순위 큐의 정렬 기준을 정의해줘야 한다. (→ 부숴야하는 벽의 개수 오름차순 정렬)

  초깃값을 세팅해주고, 우선순위큐가 빌 때까지 반복한다.

  상하좌우 4방향으로 돌면서 map[][] 을 벗어나지 않고, result[][]가 초깃값(Integer.MAX_VALUE)이라면,

→ 이동한 곳이 벽이라면(map[x][y] == 1), 기존에 부숴야하는 벽의 개수 + 1과 result[x][y] 중에 최솟값을 result[x][y]에 저장

→ 이동한 곳이 벽이 아니라면(map[x][y] == 0), 기존에 부숴야하는 벽의 개수 그대로 result[x][y]에 저장

그리고 우선순위 큐에 x, y를 추가해준다.

 

4. 결과값 출력: 모든 과정을 끝내면 result[M-1][N-1]에는 부숴야 할 최소 벽의 개수가 저장되어 있다.

.

package codingTestStudy.week4;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class B_1261 {
    static int N, M;
    static int[][] map;
    static int[][] result;	// 부숴야 하는 벽의 최소 개수 저장 
    static int[] dx = {0, 0, -1, 1};
    static int[] dy = {1, -1, 0, 0};

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

        for (int i = 0; i < M; i++) {
            String s = br.readLine();
            for (int j = 0; j < N; j++) {
                map[i][j] = s.charAt(j) - '0';

                // result[][] 초기화
                result[i][j] = Integer.MAX_VALUE;
            }
        }

        bfs(0, 0);
        System.out.println(result[M - 1][N - 1]);
    }

    static void bfs(int i, int j) {
        PriorityQueue<int[]> pq = new PriorityQueue<>((o1, o2) -> {
            // 부숴야 하는 벽 개수 오름차순 정렬 정의
            return result[o1[0]][o1[1]] - result[o2[0]][o2[1]];
        });
        pq.add(new int[]{i, j});
        result[i][j] = 0;

        while (!pq.isEmpty()) {
            int[] now = pq.poll();
            for (int k = 0; k < 4; k++) {
                int x = now[0] + dx[k];
                int y = now[1] + dy[k];

                if (x < 0 || y < 0 || x >= M || y >= N) continue;
                if (result[x][y] == Integer.MAX_VALUE) {
                    if (map[x][y] == 0)
                        result[x][y] = result[now[0]][now[1]];
                    else
                        result[x][y] = Math.min(result[x][y], result[now[0]][now[1]] + 1);
                    pq.add(new int[]{x, y});
                }
            }
        }
    }
}