글 읽기 - 파이썬 런타임오류 한번 봐주세요 ~~ ㅠㅠ

jdw0429 4년 전 0

dfs 재귀함수를 이용해서 풀었습니다.

dfs 재귀함수 안에서 경계조건까지 체크 했는데 런타임 오류가 75%에서 발생합니다.

백준님 강의자료 C++ 코드를 기반으로 작성했습니다.

다른 질문글들에 있는 반례까지 다 입력했을 때 문제가 없는데 런타임 오류가 발생합니다....

import sys
sys.setrecursionlimit(1000000000)
def dfs(a,c,i, j, n, cnt):
    dx = [-1, 1, 0 , 0] ### 4방향
    dy = [0, 0, 1, -1]
    c[i][j] = cnt 

    for k in range(4):
        next_i = i + dx[k]  ## 다음좌표
        next_j = j + dy[k]
        if (next_i >= 0) and ( next_i < n) and (next_j >= 0) and ( next_j < n):
            if (a[next_i][next_j] == 1) and (c[next_i][next_j] == False):
                dfs(a, c, next_i, next_j, n,cnt)



n = int(sys.stdin.readline()[:-1])
a = [[] for i in range(n)]
for i in range(n):
	a[i] += list(map(int, sys.stdin.readline()[:-1])) ## read map
c = [ [False for j in range(n)] for i in range(n)]    ### 방문정보 저장
   
cnt = 0  ## 단지번호
for i in range(n):
    for j in range(n):
        if a[i][j] == 1 and c[i][j] == False:
            cnt += 1
            dfs(a, c, i, j, n, cnt)

print(cnt)
if cnt != 0:
    ans = [0 for i in range(cnt+1)]            
    for i in range(n):
        for j in range(n):
            ans[c[i][j]] += 1

    ans = ans[1:]
    ans.sort()
    for i in ans:
        print(i)