프로그래머스 BFS 미로 탈출 - Level2

https://school.programmers.co.kr/learn/courses/30/lessons/159993#qna

프로그래머스

문제 설명

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

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

 

제한사항

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

문제 해결

기본적인 bfs문제 유형으로 해당 문제는 최단 경로를 탐색하는건 동일하지만 미로의 문을 열기 위해서는 무조건 레버를 먼저 당겨야 한다. 따라서 레버를 찾는 최단 경로 탐색 1회와 레버부터 출구까지 가는 최단 경로 탐색 1번 총 2번의 bfs탐색을 이용하면 쉽게 해결할 수 있는 문제이다. 여기서 주의해야 할 점은 레버를 찾는 최단경로가 없을 경우 -1을 리턴하고 종료하는 것 이외에도 레버는 찾았지만 출구를 찾지 못한 경우도 있으므로 이부분을 예외처리하는걸 신경 써야 한다.

#include <string>
#include <vector>
#include <iostream>
#include <queue>
#define MAX_SZ 100

using namespace std;
typedef long long ll;
typedef pair<ll, ll> pi;

vector<string>MAP(MAX_SZ);

pi START;
void init(vector<string>&maps){
    for(ll i=0; i<maps.size(); i++){
        MAP[i] = maps[i];
    }
}

pi find_point(char ch, vector<string>&maps){
    pi p;
    for(ll i=0; i<maps.size(); i++){
        for(ll j=0; j<maps[i].size(); j++){
            if(maps[i][j] == ch){
                return make_pair(i, j);
            }
        }
    }
    return p;
}

bool is_possible(ll ny, ll nx, ll n, ll m){
    if(ny < 0 || ny >= n)
        return false;
    if(nx < 0 || nx >= m)
        return false;
    return true;
}

ll bfs(pi s , ll n, ll m, char target){
    ll dx[4] = {0, 0, 1, -1};
    ll dy[4] = {1, -1, 0, 0};
    vector<vector<ll> >dis(n, vector<ll>(m, 0));
    queue<pi>q;
    q.push(s);
    dis[s.first][s.second] = 0;
    ll cnt = 0;
    
    while(!q.empty()){
        pi oldp = q.front(); q.pop();
        for(ll i=0; i<4; i++){
            ll ny = dy[i] + oldp.first;
            ll nx = dx[i] + oldp.second;
            if(is_possible(ny, nx, n, m) == true and MAP[ny][nx] != 'X' and dis[ny][nx] == 0){
                // 인덱스 범위 내에 있으면서 벽이 아니고 최초 방문인 경우
                dis[ny][nx] = dis[oldp.first][oldp.second] + 1;
                q.push(make_pair(ny, nx));
            }
            
        }
    } // while
    pi t = find_point(target, MAP);
    if(dis[t.first][t.second] != 0){
        cnt = dis[t.first][t.second];
        START = t;
    }
    return cnt;
}


int solution(vector<string> maps) {
    // TO DO
    init(maps);
    START = find_point('S', maps);
    ll l = 0; ll e = 0;
    l = bfs(START, maps.size(), maps[0].size(), 'L');
    if(l == 0)
        return -1;
    e = bfs(START, maps.size(), maps[0].size(), 'E');
    if(e == 0) return -1;
    return l+e;
}