[백준 21736번] 파이썬 - 헌내기는 친구가 필요해

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

시간 제한 1초, 메모리 제한 1024MB

# 조건

• 2020년에 입학한 헌내기 도연이가 있다.
• 도연이는 비대면 수업 때문에 학교에 가지 못해 학교에 아는 친구가 없었다. 드디어 대면 수업을 하게 된 도연이는 어서 캠퍼스 내의 사람들과 친해지고 싶다.
• 도연이가 다니는 대학의 캠퍼스는 N x M 크기이며 캠퍼스에서 이동하는 방법은 벽이 아닌 상하좌우로 이동하는 것이다.
  • 예를 들어, 도연이가 (x, y)에 있다면 이동할 수 있는 곳은 (x+1, y), (x, y+1), (x-1, y), (x, y-1)이다.
  • 단, 캠퍼스의 밖으로 이동할 수는 없다.
• 불쌍한 도연이를 위하여 캠퍼스에서 도연이가 만날 수 있는 사람의 수를 출력하는 프로그램을 작성해보자.

입력

• 첫째 줄에는 캠퍼스의 크기를 나타내는 두 정수 N (1<=N<=500), M (1<=M<=600)이 주어진다.
• 둘째 줄부터 N개의 줄에는 캠퍼스의 정보들이 주어진다.
• O는 빈 공간, X는 벽, I는 도연이, P는 사람이다.
• I가 한 번만 주어짐이 보장된다.

출력

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

# 접근 방법

• BFS를 사용하여 사람의 수를 구해주면 된다.
• deque를 활용하여 사람 또는 빈 공간인 경우 추가해주고, 사람이면 cnt += 1을 해준다.
• 출력시 result = 0 이면 TT를 출력해준다.
• 함수를 모듈화해주는 것이 디버깅에 편리하다는 것을 느끼고 있어서, 최대한 나눠주려고 노력중이다.

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  
from collections import deque  

def can_go(ni, nj):  
    if 0<=ni<N and 0<=nj<M and not visited[ni][nj] and arr[ni][nj] != 'X':  
        return 1  
    else:  
        return 0  

def bfs(i, j):  
    cnt = 0  
    q = deque()  
    q.append((i, j))  
    visited[i][j] = True  
    while q:  
        si, sj = q.popleft()  
        for d in range(4):  
            ni, nj = si+di[d], sj+dj[d]  
            can = can_go(ni, nj)  
            if can:  
                if arr[ni][nj] == 'P':  
                    cnt += 1  
                visited[ni][nj] = True  
                q.append((ni, nj))  
    return cnt  


N, M = map(int, input().split())  
arr = [list(input().rstrip()) for _ in range(N)]  
di, dj = [1,-1,0,0], [0,0,-1,1]  
visited = [[False] * M for _ in range(N)]  

for i in range(N):  
    for j in range(M):  
        if arr[i][j] == 'I':  
            result = bfs(i, j)  
            print('TT' if result == 0 else result)  
            break