글 읽기 - 파이썬 시간초과 질문

diziew 2년 전 0

파이썬

왜 시간초과가 날까요..

import heapq
test_case = int(input())
for __ in range(test_case):
    n,d,c = map(int,input().split())
    adj = [[] for _ in range(n+1)]
    hacked =[0]*(n+1)
    for _ in range(d):
        a,b,s = map(int,input().split())
        adj[b].append((s,a))
    queue = []
    heapq.heappush(queue,(0,0,c))
    time = 0
    while queue:
        s,time,c = heapq.heappop(queue)
        time+=s
        if hacked[c] == 0 or hacked[c] > time:
                hacked[c] = time
        for t,co in adj[c]:
                heapq.heappush(queue,(t,hacked[c] ,co))
                        
    cnt=1
    for i in hacked:
        if i!=0:
            cnt+=1
    print(cnt,max(hacked))

choko100 1년 전 0

안녕하세요, 19 번 라인을 보면 if 조건 없이 무조건 heappush 를 하게 되어있는데요. a와 b가 서로 감염시킬 수 있으면 무한루프에 걸리게 될 것 같습니다.

입력

정답

2 5

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

문제

문제

출처

ICPC

파이썬 시간초과 질문

diziew 2년 전 0

choko100 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말