[백준][BOJ][C++][2589번] 보물섬

문제

보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다. 육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.

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

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

입력

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

출력

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

 

코드

#include<iostream>
#include<queue>
#include<algorithm>
#include<memory.h>
using namespace std;
 
struct info {
    int x;
    int y;
    int t;
};
 
int X, Y;
int ans;
char map[51][51];
int visited_dfs[51][51];
int visited_bfs[51][51];
int p_x[4] = { 1,0,-1,0 };
int p_y[4] = { 0,1,0,-1 };
queue<info> q;
 
void printt() {
    for (int y = 0; y < Y; y++) {
        for (int x = 0; x < X; x++) {
            cout << map[y][x] << " ";
            //cout<<visited_bfs[y][x];
        }
        cout << endl;
    }
    cout << endl;
}
 
void bfs(int s_x, int s_y) {
    //cout << "bfs사작" << endl;
    info s;
    s.x = s_x; s.y = s_y; s.t = 0;
    visited_bfs[s.y][s.x] = 1;
    q.push(s);
    while (!q.empty()) { //남아있는 큐가 없을때까지
        int x, y, t;
        x = q.front().x; y = q.front().y; t = q.front().t;
        q.pop();
        for (int i = 0; i < 4; i++) {
            int n_x, n_y, n_t;
            n_x = x + p_x[i];
            n_y = y + p_y[i];
            if (0 > n_y && n_y >= Y && 0 > n_x && n_x >= X) continue;
            if (!visited_bfs[n_y][n_x] && map[n_y][n_x] == 'L') { //땅O,방문한곳X
                visited_bfs[n_y][n_x] = 1;
                n_t = t + 1;
                ans = max(ans, n_t);
                s.x = n_x; s.y = n_y; s.t = n_t;
                q.push(s);
            }
        }
    }
 
}
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin >> Y >> X;
    for (int y = 0; y < Y; y++) {
        string s;
        cin >> s;
        for (int x = 0; x < X; x++) {
            map[y][x] = s[x];
        }
    }
    ans = -1;
    for (int y = 0; y < Y; y++) {
        for (int x = 0; x < X; x++) {
            if (map[y][x] == 'W') continue;
            bfs(x, y);
            while (!q.empty()) q.pop();
            for (int yy = 0; yy < Y; yy++)
                for (int xx = 0; xx < X; xx++)
                    visited_bfs[yy][xx] = 0;
        }
    }
    cout << ans;
}

풀이

처음에는 DFS로 2개의 보물 위치를 잡은 다음 2개의 보물에서 마지막에 선택된 장소를 골라 BFS를 사용하여 풀이를 진행하였습니다.

 

하지만 DFS를 통해 2개의 보물 위치를 선택한다음 보물끼리의 거리를 구하는 방법은 시간초과가 발생하였고 그냥 모든 땅을 BFS로 돌렸고 그결과가 가장 큰 값을 선택하였습니다.

JAVA를 이용한 풀이법은 시간이 나는대로 풀어 올리도록 하겠습니다.

 

아래의 코드는 시간초과가 발생한 DFS코드입니다. 나중에 DFS를 사용하는 다른문제에 도움이 될 수 있다고 생각하여 코드 첨부해 드립니다.

 

틀린코드

#include<iostream>
#include<queue>
#include<algorithm>
#include<memory.h>
using namespace std;
 
struct info {
    int x;
    int y;
    int t;
};
 
int X, Y;
int ans;
char map[51][51];
int visited_dfs[51][51];
int visited_bfs[51][51];
int p_x[4] = { 1,0,-1,0 };
int p_y[4] = { 0,1,0,-1 };
queue<info> q;
 
void printt() {
    for (int y = 0; y < Y; y++) {
        for (int x = 0; x < X; x++) {
            cout << map[y][x] << " ";
            //cout<<visited_bfs[y][x];
        }
        cout << endl;
    }
    cout << endl;
}
 
void bfs(int s_x, int s_y) {
    //cout << "보물위치" << endl;
    //printt();
    int temp_ans;
    bool f;//보물발견
    f = false;
    info s;
    s.x = s_x; s.y = s_y; s.t = 0;
    q.push(s);
    while (!q.empty()) { //남아있는 큐가 없을때까지
        if (f) break;
        int x, y, t;
        x = q.front().x; y = q.front().y; t = q.front().t;
        q.pop();
        visited_bfs[y][x] = 1;
        for (int i = 0; i < 4; i++) {
            int n_x, n_y, n_t;
            n_x = x + p_x[i];
            n_y = y + p_y[i];
            if (!visited_bfs[n_y][n_x]) { //땅O,방문한곳X
                if (map[n_y][n_x] != 'L' && map[n_y][n_x] != 'X') continue;
                n_t = t + 1;
                if (map[n_y][n_x] == 'X') {//보물위치
                    f = true;
                    temp_ans = n_t;
                    ans = max(ans, temp_ans);
                    temp_ans = 0;
                    break;
                }
                s.x = n_x; s.y = n_y; s.t = n_t;
                q.push(s);
            }
        }
    }
 
}
 
void dfs(int cnt, int XX, int YY) {
    if (cnt == 2) {//보물 위치 2개 뽑기
        memset(visited_bfs, 0, sizeof(visited_bfs));
        while (!q.empty()) q.pop(); //큐 초기화
        bfs(XX, YY);
        //printt(); 
        return;
    }
 
    for (int y = YY; y < Y; y++) {
        for (int x = XX; x < X; x++) {
            if (visited_dfs[y][x] || map[y][x] == 'W') continue; // 바디이거나 이미 선택된곳이면 통과
            visited_dfs[y][x] = 1;
            map[y][x] = 'X';
            dfs(cnt + 1, x, y);
            visited_dfs[y][x] = 0;
            map[y][x] = 'L';
        }
    }
}
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin >> Y >> X;
    for (int y = 0; y < Y; y++) {
        string s;
        cin >> s;
        for (int x = 0; x < X; x++) {
            map[y][x] = s[x];
        }
    }
    ans = -1;
    dfs(0, 0, 0);
    cout << ans;
}