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

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

문제

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를 출력한다.

예제 입력 1 복사

3 5
OOOPO
OIOOX
OOOXP

예제 출력 1 복사

1

예제 입력 2 복사

3 3
IOX
OXP
XPP

예제 출력 2 복사

TT

 

 

런타임 에러

해당 문제를 풀었을 때 런타임 에러로 RecursionError 가 발생했다.

백준에 들어가서 설명을 보았다.

 

RecursionError는 재귀와 관련된 에러입니다. 가장 많이 발생하는 이유는 Python이 정한 최대 재귀 깊이보다 재귀의 깊이가 더 깊어질 때입니다.

Python이 정한 최대 재귀 깊이는 sys.getrecursionlimit()을 이용해 확인할 수 있습니다. BOJ의 채점 서버에서 이 값은 1,000으로 되어 있습니다.

이 에러를 해결하는 방법은 2가지가 있습니다.

첫 번째는 재귀 함수를 사용하지 않는 것입니다.

두 번째는 sys.setrecursionlimit()을 사용하는 것입니다. 이 함수를 사용하면, Python이 정한 최대 재귀 갚이를 변경할 수 있습니다. 소스 1의 최대 재귀 깊이를 1,000,000 정도로 크게 설정하면 런타임 에러 없이 실행이 됩니다.

import sys
sys.setrecursionlimit(10**6)

 

 

문제풀이

문제를 푸는 방법은 다른 그래프 문제와 비슷하게 완전탐색으로 풀면 된다. 나의 경우는 하던대로 dfs 로 풀었다. I 일때 dfs를 시작하여 'O' 와 'P' 일때만 이동하도록 하였다.

import sys
sys.setrecursionlimit(10**6)
input = sys.stdin.readline
def dfs(row,col):
    global campus
    global count
    global directions
    global visited
    visited[row][col] = True
    
    for x,y in directions:
        x = x + row
        y = y + col
        if 0 <= x < len(visited) and 0 <= y < len(visited[0]) :
            if not visited[x][y]:
                if campus[x][y] == 'O':
                    dfs(x,y)
                elif campus[x][y] == 'P':
                    count += 1
                    dfs(x,y)

    return

row_len , col_len = map(int,input().split())
campus = [list(input())for _ in range(row_len)]
visited = [[False for _ in range(col_len)] for _ in range(row_len)]
count = 0
directions = [(-1,0),(1,0),(0,-1),(0,1)]
for row in range(row_len):
    for col in range(col_len):
        if campus[row][col] == 'I':
            dfs(row,col)

if not count:
    print('TT')
else:
    print(count)