N×M크기의 배열로 표현되는 미로가 있다.
미로에서 1은 이동할 수 있는 칸을 나타내고, 0은 이동할 수 없는 칸을 나타낸다. 이러한 미로가 주어졌을 때, (1, 1)에서 출발하여 (N, M)의 위치로 이동할 때 지나야 하는 최소의 칸 수를 구하는 프로그램을 작성하시오. 한 칸에서 다른 칸으로 이동할 때, 서로 인접한 칸으로만 이동할 수 있다.
위의 예에서는 15칸을 지나야 (N, M)의 위치로 이동할 수 있다. 칸을 셀 때에는 시작 위치와 도착 위치도 포함한다.
#include <iostream>
#include <queue>
using namespace std;
int board[101][101];
bool visited[101][101];
int dx[4] = {1, 0, -1, 0};
int dy[4] = {0, 1, 0, -1};
int bfs(int n, int m)
{
int res = 0;
queue<pair<pair<int, int>, int> > Q;
Q.push(make_pair(make_pair(0, 0), 1));
visited[0][0] = true;
while (!Q.empty())
{
auto cur = Q.front();
Q.pop();
// 미로의 끝에 도달하면 탐색 종료
if (cur.first.first == n - 1 && cur.first.second == m - 1)
{
res = cur.second;
break;
}
for (int dir = 0; dir < 4; dir++)
{
int nx = cur.first.first + dx[dir];
int ny = cur.first.second + dy[dir];
if (nx < 0 || nx >= n || ny < 0 || ny >= m)
continue;
if (board[nx][ny] == 0)
continue;
if (visited[nx][ny])
continue;
Q.push(make_pair(make_pair(nx, ny), cur.second + 1));
visited[nx][ny] = true;
}
}
return (res);
}
int main()
{
int n, m, ans;
cin >> n >> m;
// 지도 입력받기
for (int i = 0; i < n; i++)
{
string s;
cin >> s;
for (int j = 0; j < m; j++)
{
board[i][j] = s[j] - '0';
}
}
ans = bfs(n, m);
cout << ans;
return 0;
}
가장 대표적인 BFS 문제라고 볼 수 있다.
시작점(1, 1)과 종점(N, M)이 고정적으로 주어지기 때문에 반복문 탈출 조건도 쉽게 설정할 수 있다.
한 가지 포인트는 미로를 탐색하는 최소 칸 수를 알기 위해 매 탐색마다 횟수를 추가하며 저장해야 한다는 점이다.
해당 풀이에서는 queue에 좌표와 방문한 횟수를 동시에 저장하여 기억하게끔 하였는데, 이 방법 말고도 visited 배열의 값을 증가시킴으로써 기억하는 방법도 존재한다.