글 읽기 - 파이썬 메모리 초과

whiteleo100 1년 전 0

다익스트라를 이용해서 풀었습니다.

총 3번의 다익스트라를 사용하는데... 왜 메모리 초과가 나오는 것일까요?

함수를 이용하기 때문에 함수가 끝나면 그 안에 있는 메모리는 모두 pop()되는데..

import sys,heapq
input = sys.stdin.readline
def find_load(start,v):
    visited = [800000 for _ in range(N + 1)]
    visited[start],h=0, []
    for destination, c in enumerate(edges[start]):
        visited[destination] =  c
        heapq.heappush(h,(c,destination))
    while h:
        cost, node = heapq.heappop(h)
        if node in v:
            total_cost[0] += cost
            v.remove(node)
            return True, node
        if visited[node] < cost:
            continue
        for destination, c in enumerate(edges[node]):
            if cost + c < visited[destination]:
                heapq.heappush(h,(cost+c,destination))
    return False, start
N,E = map(int, input().split())
edges = [[80000 for _ in range(N + 1)]for _ in range(N+1)]
for _ in range(E):
    a,b,c = map(int, input().split())
    edges[a][b] = min(edges[a][b], c)
    edges[b][a] = min(edges[b][a], c)
v,n,total_cost = list(map(int, input().split())),[N],[0]
l, start = find_load(1,v)
if l:
    l, start = find_load(start,v)
    if l:
        l, start = find_load(start,n)
        if l:
            print(total_cost[0])
            exit()
print(-1)

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

문제

문제

출처

ICPC

파이썬 메모리 초과

whiteleo100 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말