글 읽기 - 파이썬 런타임 에러 질문입니다..

42desix 5년 전 0

메모리를 너무 많이 잡아먹어서 생기는 문제라고들 하시던데 파이썬에서는 어떻게 사용 메모리를 줄일 수 있을까요?

import sys
import math
import heapq

def dijkstra(list_ve, edges, start_node):
    unvisiteds = []
    distances = {}

    # to-visit list
    for v in range(list_ve[0]):
        if v + 1 != start_node:
            unvisiteds.append(v + 1)

    # distance setting
    for v in range(list_ve[0]):
        if v + 1 == start_node:
            distances[v + 1] = 0
        else:
            distances[v + 1] = math.inf

    while len(unvisiteds) > 0:
        next_start_node_selector = []
        
        # update distances
        for edge in edges:
            if edge[0] == start_node:
                distances[edge[1]] = min(edge[2] + distances[start_node], distances[edge[1]])
                next_start_node_selector.append({'node':edge[1], 'cost':distances[edge[1]]})
        
        # update start node
        if next_start_node_selector == []:
            break
        else:
            start_node = heapq.nsmallest(1, next_start_node_selector, lambda s: s['cost'])[0]['node']
            unvisiteds.remove(start_node)

    return distances



if __name__ == '__main__':

    list_ve = list(map(int, sys.stdin.readline().split(' ')))
    start = int(sys.stdin.readline())

    edges = []
    for i in range(list_ve[1]):
        edges.append(list(map(int, sys.stdin.readline().split(' '))))

    distances_from_start = dijkstra(list_ve, edges, start)
    for idx in range(len(distances_from_start)):
        print(str.upper(str(distances_from_start[idx + 1])))

jh05013 5년 전 0

모든 런타임에러가 메모리 때문에 나는 건 아닙니다.

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

문제

문제

출처

ICPC

파이썬 런타임 에러 질문입니다..

42desix 5년 전 0

jh05013 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회 / 고등학교 대회

출처

대학교 대회

도움말