[프로그래머스] 미로 탈출 c++

문제 설명

1 x 1 크기의 칸들로 이루어진 직사각형 격자 형태의 미로에서 탈출하려고 합니다. 각 칸은 통로 또는 벽으로 구성되어 있으며, 벽으로 된 칸은 지나갈 수 없고 통로로 된 칸으로만 이동할 수 있습니다. 통로들 중 한 칸에는 미로를 빠져나가는 문이 있는데, 이 문은 레버를 당겨서만 열 수 있습니다. 레버 또한 통로들 중 한 칸에 있습니다. 따라서, 출발 지점에서 먼저 레버가 있는 칸으로 이동하여 레버를 당긴 후 미로를 빠져나가는 문이 있는 칸으로 이동하면 됩니다. 이때 아직 레버를 당기지 않았더라도 출구가 있는 칸을 지나갈 수 있습니다. 미로에서 한 칸을 이동하는데 1초가 걸린다고 할 때, 최대한 빠르게 미로를 빠져나가는데 걸리는 시간을 구하려 합니다.

미로를 나타낸 문자열 배열 maps가 매개변수로 주어질 때, 미로를 탈출하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 만약, 탈출할 수 없다면 -1을 return 해주세요.

제한 사항

• 5 ≤ maps의 길이 ≤ 100
  • 5 ≤ maps[i]의 길이 ≤ 100
  • maps[i]는 다음 5개의 문자들로만 이루어져 있습니다.
    • S : 시작 지점
    • E : 출구
    • L : 레버
    • O : 통로
    • X : 벽
  • 시작 지점과 출구, 레버는 항상 다른 곳에 존재하며 한 개씩만 존재합니다.
  • 출구는 레버가 당겨지지 않아도 지나갈 수 있으며, 모든 통로, 출구, 레버, 시작점은 여러 번 지나갈 수 있습니다.

입출력 예

maps	result
["SOOOL","XXXXO","OOOOO","OXXXX","OOOOE"]	16
["LOOXS","OOOOX","OOOOO","OOOOO","EOOOO"]	-1

풀이

최단 거리 계산은 bfs알고리즘으로 풀 수 있습니다.

시작 지점, 종료 지점, 레버 지점 3가지가 있습니다.

시작 지점과 종료 지점 각각 기준으로 거리 계산을 합니다.

시작 지점 기준으로 계산은 board[0][ ][ ], 종료 지점 기준으로 계산은 board[1][ ][ ]에 저장합니다.

마지막으로 레버 지점의 board[0][x][y] + board[1][x][y] 더하면 최단 거리를 구할 수 있습니다.

각각 계산했기 때문에 -2를 해야 합니다.

 

레버 지점 계산 결과가 0이면 시작 지점과 종료 지점에서 이동을 못하는 경우이기 때문에 -1을 반환하는 예외처리를 추가합니다.

#include <string>
#include <vector>
#include <queue>

using namespace std;

int dist[6] = { 1, 0, -1, 0, 1 };
int board[2][101][101];
queue<pair<int, int>> Q;
int w, h;

void bfs(int idx, vector<string>& maps) {
    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 + dist[i];
            int ny = y + dist[i + 1];
            if (nx >= 0 && nx < h && ny >= 0 && ny < w && board[idx][nx][ny] == 0) {
                if (maps[nx][ny] != 'X') {
                    board[idx][nx][ny] = board[idx][x][y] + 1;
                    Q.push({ nx, ny });
                }
            }
        }
    }
}

int solution(vector<string> maps) {

    h = maps.size();
    w = maps[0].size();

    int j;
    for (int i = 0; i < h; i++) {
        for (j = 0; j < w; j++) {
            if (maps[i][j] == 'S') {
                board[0][i][j] = 1;
                Q.push({ i, j });
                bfs(0, maps);
                break;
            }
        }
        if (j < w) {
            break;
        }
    }

    for (int i = 0; i < h; i++) {
        for (j = 0; j < w; j++) {
            if (maps[i][j] == 'E') {
                board[1][i][j] = 1;
                Q.push({ i, j });
                bfs(1, maps);
                break;
            }
        }
        if (j < w) {
            break;
        }
    }
     
     int x = 0, y = 0;
     for (int i = 0; i < h; i++) {
        for (j = 0; j < w; j++) {
            if (maps[i][j] == 'L') {               
                x = i, y = j;
                break;
            }
        }
        if (j < w) {
            break;
        }
    }

    if(board[0][x][y] == 0 || board[1][x][y] == 0){
        return -1;
    }
    
    int answer = board[0][x][y] + board[1][x][y] - 2;
    
    return answer;
}