[Python] 백준 21736번 - 헌내기는 친구가 필요해 (실버 2)

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

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

 

혼자 힘으로 풀었는가? O

알고리즘 분류
 - 그래프 이론
 - 그래프 탐색
 - 너비 우선 탐색(BFS)
 - 깊이 우선 탐색(DFS)

 

문제

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

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

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

입력

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

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

출력

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

---

어렵지 않은 문제다.

 

DFS와 BFS 두 가지 방법 모두 사용해서 풀 수 있다.

 

BFS

 

import sys
input = sys.stdin.readline
from collections import deque

n, m = map(int, input().split())
data = []
x, y = -1, -1
for i in range(n):
    data.append(list(input().rstrip()))
    if x == -1 and 'I' in data[i]:
        for j in range(len(data[i])):
            if data[i][j] == 'I':
                x, y = i, j
                
q = deque()
q.append((x, y))

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

visited = [ [0 for _ in range(m)] for _ in range(n)]
visited[x][y] = 1
cnt = 0
while q:
    x, y = q.popleft()
    
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        if 0 <= nx < n and 0 <= ny < m and visited[nx][ny] == 0:
            if data[nx][ny] == 'X':
                continue
            if data[nx][ny] == 'P':
                cnt += 1
            q.append((nx, ny))
            visited[nx][ny] = 1
            
if cnt == 0:
    print('TT')
else:
    print(cnt)

 

 

DFS

 

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**6)

n, m = map(int, input().split())
data = []
x, y = -1, -1
for i in range(n):
    data.append(list(input().rstrip()))
    if x == -1 and 'I' in data[i]:
        for j in range(len(data[i])):
            if data[i][j] == 'I':
                x, y = i, j

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

visited = [ [0 for _ in range(m)] for _ in range(n)]
visited[x][y] = 1
cnt = 0

def dfs(x, y):
    global cnt
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        if 0 <= nx < n and 0 <= ny < m and visited[nx][ny] == 0:
            if data[nx][ny] == 'X':
                continue
            if data[nx][ny] == 'P':
                cnt += 1
            visited[nx][ny] = 1
            dfs(nx, ny)

dfs(x, y)
            
if cnt == 0:
    print('TT')
else:
    print(cnt)

 

어렵지 않게 풀 수 있었다.