글 읽기 - 파이썬 런타임 오류 질문드립니다

ssango 4년 전 0

18퍼정도 진행되다가 런타임 오류가 나는데 .. 이거는 메모리를 많이 잡고있어서 그런건가요??

도움요청드립니다

import copy as cp

arr =[int(i)for i in input().split(' ')]
N =arr[0]
L =arr[1]
M = arr[2]
matrix =[]

for i in range(N):
    matrix.append([int(i)for i in input().split(' ')])

count = 0
total =0

def dfs(a,b,visit_num,visit,arr):
    global count
    global total
    if visit[a][b] == visit_num:
        return
    total += arr[a][b]
    count+=1
    visit[a][b] =visit_num
    if a+1< N and abs(arr[a+1][b] - arr[a][b]) >= L and abs(arr[a+1][b] -arr[a][b])<=M :
        dfs(a+1,b,visit_num,visit,arr)
    if b+1 < N and abs(arr[a][b+1] -arr[a][b])>=L and abs(arr[a][b+1] -arr[a][b])<=M :
        dfs(a,b+1,visit_num,visit,arr)
    if a-1>=0 and abs(arr[a-1][b] -arr[a][b]) >= L and abs(arr[a-1][b] -arr[a][b])<=M :
        dfs(a-1,b,visit_num,visit,arr)
    if b-1 >=0 and abs(arr[a][b-1] -arr[a][b])>=L and abs(arr[a][b-1] -arr[a][b])<=M :
        dfs(a,b-1,visit_num,visit,arr)
    return


def solve(human_shift,arr):
    global count
    global total
    total_compare = cp.deepcopy(arr)
    visit_num = 0
    visit = [[-1] * N for i in range(N)]
    for i in range(N):
        for j in range(N):
            dfs(i,j,visit_num,visit,arr)
            if count != 0:
                avg = int(total/count)
            else:
                avg = 0
            for k in range(N):
                for l in range(N):
                    if visit[k][l] == visit_num:
                        if avg != 0:
                            total_compare[k][l] = avg
            visit_num+=1
            count=0
            total=0

    if arr != total_compare:
        return solve(human_shift+1,total_compare)
    else:
        return human_shift
print(solve(0,matrix))

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

파이썬 런타임 오류 질문드립니다

ssango 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말