[BOJ #1926] 그림

문제

어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자. 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다. 그림의 넓이란 그림에 포함된 1의 개수이다.

 

입력

첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)

 

출력

첫째 줄에는 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라. 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.

 

내가 쓴 코드

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**6)
di = [0, 0, 1, -1]
dj = [1, -1, 0, 0]

def dfs(a, b):
    global area
    for k in range(4):
        ni = a + di[k]
        nj = b + dj[k]
        if 0<=ni<n and 0<=nj<m and arr[ni][nj] and not visited[ni][nj]:
            visited[ni][nj] = True
            area += 1
            dfs(ni, nj)

n, m = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(n)]
visited = [[False] * m for _ in range(n)]
# 그림 수, 가장 넓은 그림의 넓이
# 그림 수 = 0 -> 넓이 = 0
grim = 0
max_area = 0

for i in range(n):
    for j in range(m):
        if arr[i][j] and not visited[i][j]:
            visited[i][j] = True
            area = 1
            dfs(i, j)
            grim += 1
            max_area = max(area, max_area)

print(grim)
print(max_area)

 

제출 결과

머지 틀린게 없는데... 하다가 결국 알아낸 실수는.. 처음 이중 for문에서 내가 방문처리를 안한 것....... 암쏘바보^^

and recursion error는 setrecursionlimit 을 설정하니 해결되었고, 메모리 초과는 pypy3 말고 python3으로 돌리니까 해결되었당...ㅎㅎ

결론 = 난 바보다..^^;;