알고리즘/BOJ

[BOJ] 2206. 벽 부수고 이동하기

재담 2022. 2. 27. 00:26

문제 원본 : https://www.acmicpc.net/problem/2206

 

2206번: 벽 부수고 이동하기

N×M의 행렬로 표현되는 맵이 있다. 맵에서 0은 이동할 수 있는 곳을 나타내고, 1은 이동할 수 없는 벽이 있는 곳을 나타낸다. 당신은 (1, 1)에서 (N, M)의 위치까지 이동하려 하는데, 이때 최단 경로

www.acmicpc.net

import java.io.*;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n, m;

        StringTokenizer st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());

        char[][] map = new char[n][m];
        for (int i = 0; i < n; ++i) {
            map[i] = br.readLine().toCharArray();
        }

        boolean[][][] v = new boolean[n][m][2];
        int[][] dist = new int[n][m];
        int[][] dir = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};
        Queue<Pos> q = new LinkedList<>();
        q.add(new Pos(0, 0, 0));

        dist[0][0] = 1;
        while (!q.isEmpty()) {
            Pos p = q.poll();
            for (int i = 0; i < 4; ++i) {
                int ny = p.y + dir[i][0];
                int nx = p.x + dir[i][1];

                if (ny >= 0 && nx >= 0 && ny < n && nx < m) {
                    if (map[ny][nx] == '0' && !v[ny][nx][p.wall]) {
                        q.add(new Pos(ny, nx, p.wall));
                        dist[ny][nx] = dist[p.y][p.x] + 1;
                        v[ny][nx][p.wall] = true;
                    } else if (map[ny][nx] == '1' && p.wall == 0 && !v[ny][nx][1]) {
                        q.add(new Pos(ny, nx, 1));
                        dist[ny][nx] = dist[p.y][p.x] + 1;
                        v[ny][nx][1] = true;
                    }

                    if (ny == n - 1 && nx == m - 1) {
                        bw.write(Integer.toString(dist[ny][nx]));
                        bw.close();
                        br.close();
                        return;
                    }
                }
            }
        }

        if (dist[n - 1][m - 1] == 0) {
            bw.write("-1");
        } else {
            bw.write("1");
        }

        bw.close();
        br.close();
    }

    public static class Pos {
        public int y;
        public int x;
        public int wall;

        public Pos(int y, int x, int wall) {
            this.y = y;
            this.x = x;
            this.wall = wall;
        }
    }
}
  • 일반적인 BFS 풀이법으로는 풀 수 없는 문제였다.
  • 벽을 한 번 부술 수 있기 때문에 전 단계에서 벽을 부순 상태도 같이 저장하고 있어야 했다.
  • 이 문제도 아이디어를 다른 블로그에서 참고했다.
  • 골드 4 티어인데 풀이법을 모른다면 체감 난이도는 골드 1쯤 되는 것 같다.

'알고리즘 > BOJ' 카테고리의 다른 글

[BOJ] 1074. Z  (0) 2022.02.28
[BOJ] 1600. 말이 되고픈 원숭이  (0) 2022.02.27
[BOJ] 1406. 에디터  (0) 2022.02.26
[BOJ] 5639. 이진 검색 트리  (0) 2022.02.25
[BOJ] 14003. 가장 긴 증가하는 부분 수열 5  (0) 2022.02.24