백준 3055 탈출 (C++)

3055번: 탈출

bfs를 이용한 문제이다. 특이한 점이라면 bfs를 통해 좌표를 이동하는 점이 여러 개라는 것이다. 맵을 입력받을 때 돌, 시작 지점과 굴의 좌표를 먼저 저장해주었다. 고슴도치의 위치와 물의 좌표를 큐에 넣을 때 물의 경우 카운트를 -1로 넣어주고 bfs를 돌려주었다. 만약 카운트가 -1이 아닐 경우 고슴도치의 좌표를 의미하므로 이를 통해 고슴도치가 굴에 도달했는 지 확인해주었다. 어렵지 않게 풀 수 있었다.

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

typedef pair<int, int> pii;

int R, C;
pii S, D;
char A[50][50];
vector<pii> water;
bool check[50][50];
int result = -1;
int dy[4] = { -1,1,0,0 };
int dx[4] = { 0,0,-1,1 };

void bfs() {
    queue<pair<pii, int>> q;

    q.push({ { S.first, S.second },0 });
    check[S.first][S.second] = true;

    for (int i = 0; i < water.size(); i++) {
        q.push({ { water[i].first, water[i].second }, -1 });
    }

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

        if (count != -1 && A[y][x] == 'D') {
            result = count;
            break;
        }

        q.pop();

        for (int i = 0; i < 4; i++) {
            int ny = y + dy[i];
            int nx = x + dx[i];

            if (ny < 0 || nx < 0 || ny >= R || nx >= C) continue;
            if (A[ny][nx] == 'X') continue;
            if (A[ny][nx] == '*') continue;

            if (A[y][x] == '*') {
                if (A[ny][nx] == 'D') continue;

                q.push({ {ny,nx},count });
                A[ny][nx] = '*';
            }
            else {
                if (check[ny][nx]) continue;

                q.push({ { ny,nx } ,count + 1 });
                check[ny][nx] = true;
            }
        }
    }
}

void solution() {
    bfs();

    if (result == -1) {
        cout << "KAKTUS";
    }
    else {
        cout << result;
    }
}

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

    cin >> R >> C;

    for (int i = 0; i < R; i++) {
        for (int j = 0; j < C; j++) {
            cin >> A[i][j];

            if (A[i][j] == '*') {
                water.push_back({ i,j });
            }
            else if (A[i][j] == 'S') {
                S = { i,j };
            }
            else if (A[i][j] == 'D') {
                D = { i,j };
            }
        }
    }

    solution();

    return 0;
}