[Baekjoon/C++] 21736번 - 헌내기는 친구가 필요해

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

도연이가 다니는 대학의 캠퍼스는 N × 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>
using namespace std;

int N, M, cnt = 0;
char arr[600][600];
bool check[600][600];

const int dx[4] = { 0, 0, 1, -1 };
const int dy[4] = { 1, -1, 0, 0 };

void dfs(int x, int y);

void dfs(int x, int y) {
    if (x < 0 || x >= N || y < 0 || y >= M) return;

    check[x][y] = true;

    if (arr[x][y] == 'X') return;
    if (arr[x][y] == 'P') cnt++;

    for (int i = 0; i < 4; i++)
        if (!check[x + dx[i]][y + dy[i]]) dfs(x + dx[i], y + dy[i]);
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    cin >> N >> M;

    int sX = -1, sY = 0;

    for (int i = 0; i < N; i++) {
        cin >> arr[i];

        if (sX == -1) {
            for (int j = 0; j < M; j++) {
                if (arr[i][j] == 'I') {
                    sX = i;
                    sY = j;

                    break;
                }
            }
        }
    }

    dfs(sX, sY);

    if (cnt == 0) cout << "TT\n";
    else cout << cnt << '\n';

    return 0;
}

 

void dfs(int x, int y) {
    if (x < 0 || x >= N || y < 0 || y >= M) return;

    check[x][y] = true;

    if (arr[x][y] == 'X') return;
    if (arr[x][y] == 'P') cnt++;

    for (int i = 0; i < 4; i++)
        if (!check[x + dx[i]][y + dy[i]]) dfs(x + dx[i], y + dy[i]);
}

캠퍼스를 순회하는 코드이다. 방문한 칸이 X이거나 캠퍼스의 범위를 넘어가면 return을 호출하고, P라면 cnt를 하나 증가시킨다. 방문한 칸을 또 방문하는 것을 막기 위해 check[][]에 방문한 칸을 표시하도록 했다.