[백준] 21736

헌내기는 친구가 필요해

문제

2020년에 입학한 헌내기 도연이가 있다. 도연이는 비대면 수업 때문에 학교에 가지 못해 학교에 아는 친구가 없었다. 드디어 대면 수업을 하게 된 도연이는 어서 캠퍼스 내의 사람들과 친해지고 싶다.

도연이가 다니는 대학의 캠퍼스는 N x M 크기이며 캠퍼스에서 이동하는 방법은 벽이 아닌 상하좌우로 이동하는 것이다. 예를 들어, 도연이가 (x, y)에 있다면 이동할 수 있는 곳은 (x+1, y), (x, y+1), (x-1, y), (x, y-1)이다. 단, 캠퍼스의 밖으로 이동할 수는 없다.

불쌍한 도연이를 위하여 캠퍼스에서 도연이가 만날 수 있는 사람의 수를 출력하는 프로그램을 작성해보자.

입력

첫째 줄에는 캠퍼스의 크기를 나타내는 두 정수 N (1 ≤ N ≤ 600), M (1 ≤ M ≤ 600)이 주어진다.

둘째 줄부터 N개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.

출력

첫째 줄에 도연이가 만날 수 있는 사람의 수를 출력한다. 단, 아무도 만나지 못한 경우 TT를 출력한다.

풀이

#include <iostream>
#include <queue>
#include <string>

bool bfs_visited[601][601];
int arr[601][601];
std::pair<int, int> startPos;
std::queue<std::pair<int, int> > que;
int dir[2][4] = {{0, 1, 0, -1}, {1, 0, -1, 0}};
int cnt;

void bfs(int n, int m) {
    que.push(startPos);
    bfs_visited[startPos.first][startPos.second] = true;

    while (!que.empty()) {
        std::pair<int, int> current = que.front();
        que.pop();

        for (int i = 0; i < 4; i++) {
            int coordY = current.first + dir[0][i];
            int coordX = current.second + dir[1][i];
            if (coordX < 1 || coordY < 1 || coordX > m || coordY > n)
                continue;
            if (bfs_visited[coordY][coordX])
                continue;
            if (arr[coordY][coordX] == 2)
                continue;
            if (arr[coordY][coordX] == 1)
                cnt++;
            bfs_visited[coordY][coordX] = true;
            que.push(std::make_pair(coordY, coordX));
        }
    }
}

int main(void) {
    int n, m;
    std::string temp;
    std::cin >> n >> m;

    for (int i = 1; i <= n; i++) {
        std::cin >> temp;
        for (int j = 0; j < m; j++) {
            if (temp[j] == 'I') {
                startPos.first = i;
                startPos.second = j + 1;
            }
            if (temp[j] == 'P')
                arr[i][j + 1] = 1;
            if (temp[j] == 'X')
                arr[i][j + 1] = 2;
        }
    }

    bfs(n, m);
    if (cnt == 0) std::cout << "TT";
    else std::cout << cnt;
}

반성회

DFS로 풀었다가 뭐가 잘못된건지 자꾸 이상하게 나와서 BFS로 고쳤는데 이상한데서 실수한거였음