[2589] 보물섬(C++)

문제

보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다.

각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다.

육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.

 

예를 들어 위와 같이 지도가 주어졌다면 보물은 아래 표시된 두 곳에 묻혀 있게 되고, 이 둘 사이의 최단 거리로 이동하는 시간은 8시간이 된다.

보물 지도가 주어질 때, 보물이 묻혀 있는 두 곳 간의 최단 거리로 이동하는 시간을 구하는 프로그램을 작성하시오.

 

 

입력

첫째 줄에는 보물 지도의 세로의 크기와 가로의 크기가 빈칸을 사이에 두고 주어진다.

이어 L과 W로 표시된 보물 지도가 아래의 예와 같이 주어지며, 각 문자 사이에는 빈 칸이 없다. 보물 지도의 가로, 세로의 크기는 각각 50이하이다.

 

 

출력

첫째 줄에 보물이 묻혀 있는 두 곳 사이를 최단 거리로 이동하는 시간을 출력한다.

 

 

알고리즘

• 브루트포스
• BFS

 

 

해결 방법

맵의 크기가 50x50으로 최대 2500이므로 브루트포스로 해결할 수 있는 문제.

입력을 받고 칸이 육지인 곳에서 전부 bfs를 수행하면 된다.

 

각 칸에서 상화좌우 네 방향으로 추가 탐색을 해야 해서 시간 초과가 나지 않을까 싶지만 모든 칸이 육지라고 해도 이미 지나간 곳은 다시 가지 않기 때문에 그냥 최대 2500칸을 탐색하는 셈이라서 1초 안에 충분히 해결 가능하다.

 

bfs 문제는 각 칸까지의 경로 길이를 저장하는 방법이 다양한데

 

visited의 역할을 단순히 방문했다/안 했다가 아니라 방문을 했다면 몇 번째 만에 방문한 건지 저장하는 용도로 써도 되고,

아니면 visited 이외에 추가로 경로만 저장할 배열을 또 선언해도 되고,

애초에 큐에 저장할 때 좌표와 이동 거리를 같이 저장해주는 방법도 있다.

 

나는 이 중 추가 배열인 path를 선언하는 방식으로 문제를 해결했다!

 

 

내 코드

// [2589] 보물섬
#include <iostream>
#include <queue>
#include <cstring>
using namespace std;

bool visited[50][50];
int h, w, ans, path[50][50];
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
string map[50];

void input()
{
    cin >> h >> w;
    for(int i = 0; i < h; i++) {
        cin >> map[i];
    }
}

void bfs(int sx, int sy)
{
    memset(path, 0, sizeof(path));
    memset(visited, false, sizeof(visited));
    queue<pair<int, int>> q;

    visited[sx][sy] = true;
    q.push({sx, sy});

    while(!q.empty()) {
        int x = q.front().first;
        int y = q.front().second;
        q.pop();

        for(int i = 0; i < 4; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if(nx < 0 || nx >= h || ny < 0 || ny >= w) continue;
            if(visited[nx][ny] || map[nx][ny] == 'W') continue;

            q.push({nx, ny});
            visited[nx][ny] = true;
            path[nx][ny] = path[x][y] + 1;
            ans = max(ans, path[nx][ny]);
        }
    }
}

void solve()
{
    for(int i = 0; i < h; i++) {
        for(int j = 0; j < w; j++) {
            if(map[i][j] == 'L') {
                bfs(i, j);
            }
        }
    }
}

int main(void)
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    input();
    solve();
    cout << ans << '\n';

    return 0;
}