[백준] 21736번 : 헌내기는 친구가 필요해 (C++)

21736번 : 헌내기는 친구가 필요해

---

 

문제)

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를 출력한다.

 

 

 

 

 

풀이)

 

// 21736. 헌내기는 친구가 필요해
#include <iostream>
#include <vector>
#include <queue>
 
using namespace std;
 
int n, m;
vector<string> vec;
vector<vector<bool>> visited;
 
pair<int, int> s;
int cnt;
 
int dx[] = { -1, 0, 1, 0 };
int dy[] = { 0, -1, 0, 1 };
 
void bfs()
{
    queue<pair<int, int>>q;
    visited[s.first][s.second] = 1;
    q.push(s);
 
    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 + dx[i];
            int ny = y + dy[i];
 
            if (0 > nx || nx >= n) continue;
            if (0 > ny || ny >= m) continue;
            if (vec[nx][ny] == 'X') continue;
            if (visited[nx][ny]) continue;
            if (vec[nx][ny] == 'P') cnt++;
 
            visited[nx][ny] = 1;
            q.push(make_pair(nx, ny));
        }
    }
}
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
 
    cin >> n >> m;
    visited.resize(n, vector<bool>(m, 0));
 
    string temp;
 
 
    for (int i = 0; i < n; i++)
    {
        cin >> temp;
        vec.push_back(temp);
        for (int j = 0; j < m; j++)
        {
            if (temp[j] == 'I')
            {
                s.first = i;
                s.second = j;
                break;
            }
        }
    }
 
    bfs();
    if (cnt)
    {
        cout << cnt;
    }
    else cout << "TT";
 
    return 0;
}
 

---

출처 : https://www.acmicpc.net/problem/21736 

21736번: 헌내기는 친구가 필요해