[백준] 21736 헌내기는 친구가 필요해

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

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

문제

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

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

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

입력

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

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

출력

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

---

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;

int N, M, ret = 0;
pair<int, int> doyun;

int main()
{
	cin >> N >> M;
	vector<vector<char>> campus(N, vector<char>(M, 'X'));
	for (int y = 0; y < N; y++)
	{
		for (int x = 0; x < M; x++)
		{
			cin >> campus[y][x];
			if (campus[y][x] == 'I')
				doyun = make_pair(y, x);
		}
	}

	//우 좌 상 하
	int _y[] = { 0,0,-1,1 };
	int _x[] = { 1,-1,0,0 };

	vector<vector<int>> visit(N, vector<int>(M, 0));
	queue<pair<int, int>> q;
	q.push(doyun);
	visit[q.front().first][q.front().second] = 1;

	while (!q.empty())
	{
		int sy = q.front().first, sx = q.front().second;
		for (int dir = 0; dir < 4; dir++)
		{
			int ny = sy + _y[dir], nx = sx + _x[dir];
			if (ny >= N || ny < 0 || nx >= M || nx < 0)continue;
			if (visit[ny][nx] || campus[ny][nx] == 'X') continue;
			if (campus[ny][nx] == 'P') ret++;
			q.push(make_pair(ny, nx));
			visit[ny][nx] = 1;
		}
		q.pop();
	}

	if (!ret) cout << "TT" << endl;
	else cout << ret << endl;


}