2206번: 벽 부수고 이동하기
N×M의 행렬로 표현되는 맵이 있다. 맵에서 0은 이동할 수 있는 곳을 나타내고, 1은 이동할 수 없는 벽이 있는 곳을 나타낸다. 당신은 (1, 1)에서 (N, M)의 위치까지 이동하려 하는데, 이때 최단 경로
www.acmicpc.net
(0, 0) ~ (N - 1, M - 1)까지 bfs로 순회하는데 벽을 한 개 까지 부수고 이동할 수 있습니다.
저는 visit배열을 3차원배열로 선언하여 (x, y, broke) => 벽을 broke번 부수고 x, y에 방문한 경우를 체크하였습니다.
벽을 이미 부순 경우에는 다음 벽을 부술 수 없고, 벽을 부수지 않은 경우에만 다음 벽을 부술 수 있습니다.
(N - 1, M - 1)에 도달하면 cnt값을 출력, 큐가 비어있음에도 도달하지 못하면 -1을 출력합니다.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static StringTokenizer st;
static Queue<int[]> q = new LinkedList<>();
static int direct[][] = { { 0, 1 }, { 1, 0 }, { 0, -1 }, { -1, 0 } };
static boolean visit[][][];
static char list[][];
static int N, M, ans = -1;
static void bfs() {
q.add(new int[] { 0, 0, 1, 0 });
visit[0][0][0] = true;
while (!q.isEmpty()) {
int x = q.peek()[0];
int y = q.peek()[1];
int cnt = q.peek()[2];
int broke = q.poll()[3];
if (x == N - 1 && y == M - 1) {
ans = cnt;
break;
}
for (int i = 0; i < 4; i++) {
int nx = x + direct[i][0];
int ny = y + direct[i][1];
if (nx < 0 || ny < 0 || nx >= N || ny >= M)
continue;
if (list[nx][ny] == '1' && broke == 1)
continue;
if (list[nx][ny] == '1') {
if (visit[nx][ny][1])
continue;
q.add(new int[] { nx, ny, cnt + 1, 1 });
visit[nx][ny][1] = true;
} else {
if (visit[nx][ny][broke])
continue;
q.add(new int[] { nx, ny, cnt + 1, broke });
visit[nx][ny][broke] = true;
}
}
}
System.out.println(ans);
}
static void input() throws Exception {
st = new StringTokenizer(br.readLine());
N = Integer.parseInt(st.nextToken());
M = Integer.parseInt(st.nextToken());
list = new char[N][];
visit = new boolean[N][M][2];
for (int i = 0; i < N; i++) {
st = new StringTokenizer(br.readLine());
list[i] = st.nextToken().toCharArray();
}
}
public static void main(String[] args) throws Exception {
input();
bfs();
}
}
|
cs |
'algorithm > bfs' 카테고리의 다른 글
| boj 16236 아기 상어 (0) | 2021.02.14 |
|---|---|
| boj 10026 적록색약 (0) | 2021.02.13 |
| boj 14502 연구소 (0) | 2021.02.12 |
| boj 20304 비밀번호 제작 (0) | 2021.02.08 |
| boj 2606 바이러스 (0) | 2021.02.03 |