문제

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

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

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

import sys

input = lambda: sys.stdin.readline().strip()

from collections import deque

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

n, m = map(int, input().split())
graph = []
x, y = 0, 0
for i in range(n):
  cam = list(input())
  graph.append(cam)
  if "I" in cam:
    y = i
    x = cam.index("I")

count = [0]


def bfs(x, y):
  queue = deque()
  queue.append((x, y))
  while queue:
    x, y = queue.popleft()
    visited[y][x] = True
    for i in range(4):
      nx, ny = x + dx[i], y + dy[i]
      if nx < 0 or ny < 0 or nx >= m or ny >= n:
        continue
      if visited[ny][nx] == False and graph[ny][nx] != "X":
        visited[ny][nx] = True
        if graph[ny][nx] == "P":
          count[0] += 1
        queue.append((nx, ny))
  if count[0] == 0:
    count[0] = "TT"


visited = [[False] * m for _ in range(n)]

bfs(x, y)

print(count[0])

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