[Silver II] 헌내기는 친구가 필요해 - 21736 python

[Silver II] 헌내기는 친구가 필요해 - 21736

문제 링크

성능 요약

메모리: 135696 KB, 시간: 240 ms

분류

너비 우선 탐색, 깊이 우선 탐색, 그래프 이론, 그래프 탐색

제출 일자

2025년 2월 20일 16:35:31

문제 설명

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

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

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

입력

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

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

출력

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

---

DFS 사용해서 풀 수 있지만 그러면 시간초과가 나더라 그래서 BFS를 사용해서 풀어보았다.

from collections import deque
import sys

N, M = map(int, sys.stdin.readline().split())
campus = []
start = []

for i in range(N):
    A = list(sys.stdin.readline().rstrip())
    for j in range(M):
        if A[j] == "I":
            start = [i, j]
    campus.append(A)

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
visited = [[0]*M for _ in range(N)]
cnt = 0

queue = deque([start])
visited[start[0]][start[1]] = 1

while queue:
    x, y = queue.popleft()
    # print(x, y)
    # print(queue)
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]

        if 0 <= nx < N and 0 <= ny < M and not visited[nx][ny]:
            if campus[nx][ny] != "X":
                queue.append([nx, ny])
                visited[nx][ny] = 1
                if campus[nx][ny] == "P":
                    cnt += 1

print("TT" if cnt == 0 else cnt)