forked from mpfeifer1/Kattis
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdungeon.cpp
96 lines (83 loc) · 2.59 KB
/
dungeon.cpp
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include <bits/stdc++.h>
using namespace std;
int x, y, z;
struct loc {
int x, y, z;
};
bool inrange(int a, int b, int c) {
return (a >= 0 && b >= 0 && b >= 0 && a < x && b < y && c < z);
}
int main() {
ios::sync_with_stdio(false);
cin.tie(NULL);
while(cin >> x >> y >> z) {
loc s, e;
if(x == 0 || y == 0 || z == 0) {
break;
}
vector<vector<vector<int>>> v(x);
for(auto& i : v) {
i.resize(y, vector<int>(z, -2));
}
for(int i = 0; i < x; i++) {
for(int j = 0; j < y; j++) {
for(int k = 0; k < z; k++) {
char c;
cin >> c;
if(c == '#') {
v[i][j][k] = -1;
}
if(c == 'S') {
s.x = i;
s.y = j;
s.z = k;
}
if(c == 'E') {
e.x = i;
e.y = j;
e.z = k;
}
}
}
}
// BFS
queue<loc> q;
q.push(s);
v[s.x][s.y][s.z] = 0;
while(!q.empty()) {
loc c = q.front();
q.pop();
// Try 6 positions
if(inrange(c.x-1,c.y,c.z) && v[c.x-1][c.y][c.z] == -2) {
v[c.x-1][c.y][c.z] = v[c.x][c.y][c.z] + 1;
q.push({c.x-1,c.y,c.z});
}
if(inrange(c.x+1,c.y,c.z) && v[c.x+1][c.y][c.z] == -2) {
v[c.x+1][c.y][c.z] = v[c.x][c.y][c.z] + 1;
q.push({c.x+1,c.y,c.z});
}
if(inrange(c.x,c.y-1,c.z) && v[c.x][c.y-1][c.z] == -2) {
v[c.x][c.y-1][c.z] = v[c.x][c.y][c.z] + 1;
q.push({c.x,c.y-1,c.z});
}
if(inrange(c.x,c.y+1,c.z) && v[c.x][c.y+1][c.z] == -2) {
v[c.x][c.y+1][c.z] = v[c.x][c.y][c.z] + 1;
q.push({c.x,c.y+1,c.z});
}
if(inrange(c.x,c.y,c.z-1) && v[c.x][c.y][c.z-1] == -2) {
v[c.x][c.y][c.z-1] = v[c.x][c.y][c.z] + 1;
q.push({c.x,c.y,c.z-1});
}
if(inrange(c.x,c.y,c.z+1) && v[c.x][c.y][c.z+1] == -2) {
v[c.x][c.y][c.z+1] = v[c.x][c.y][c.z] + 1;
q.push({c.x,c.y,c.z+1});
}
}
if(v[e.x][e.y][e.z] < 0) {
cout << "Trapped!" << endl;
}
else {
cout << "Escaped in " << v[e.x][e.y][e.z] << " minute(s)." << endl;
}
}
}