글 읽기 - 10217 파이썬 질문입니다.

ris33 3년 전 0

메모리 초과가 나옵니다.

heap에 들어가는 튜플도 최소로 넣는것같은데

해결하려면 어떻게 해야할까요?

import sys
import heapq
input=sys.stdin.readline
INF=int(1e9)
t=int(input().strip())
for _ in range(t):
       n,m,k=map(int,input().split())
       data=[[] for i in range(n+1)]
       for _ in range(k):
              u,v,c,d=map(int,input().split())
              data[u].append((v,c,d))
       distance=[INF]*(n+1)
       distance[1]=0
       q=[]
       heapq.heappush(q,(0,0,1))
       while q:
              km,mki,index=heapq.heappop(q)
              for i in data[index]:
                     cost=km+i[2]
                     sibal=mki+i[1]
                     if sibal<=m:
                            heapq.heappush(q,(cost,sibal,i[0]))
                            if cost<distance[i[0]]:
                                   distance[i[0]]=cost

       if distance[n]==INF:
              print('Poor KCM')
       else:
              print(distance[n])

02q09q 1년 전 0

저 진짜 왜 그러는지 몰라서 여러 코드 비교하면서 해보니까

위에 for _ in range(int(input())대신 def main(): 함수로 바꿔주시고

함수를 for _in range(int(input()): 안에서 호출하니까 되더라고요,.?한번 해보세요

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

문제

문제

출처

ICPC

10217 파이썬 질문입니다.

ris33 3년 전 0

02q09q 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말