Submission #73633683


Source Code Expand

from heapq import heappop, heappush

N, M, T = [*map(int, input().split())]
T -= 1

edges = [tuple(map(int, input().split())) for _ in range(M)]
edges = [(A - 1, B - 1, C) for A, B, C in edges]

adj = [[] for _ in range(N)]
for u, v, w in edges:
    adj[u].append((v, w))
    adj[v].append((u, w))

INF = 10**20


def dijk(s: int) -> list[int]:
    dis = [INF] * N
    vis = [False] * N
    pq = []

    dis[s] = 0
    heappush(pq, (0, s))

    while pq:
        d, u = heappop(pq)

        if vis[u]:
            continue

        vis[u] = True

        for v, w in adj[u]:
            if not vis[v] and dis[v] > d + w:
                dis[v] = d + w
                heappush(pq, (dis[v], v))

    return dis


dis = dijk(0)

if dis[T] != INF:
    print(2 * dis[T])
else:
    print(-1)

Submission Info

Submission Time
Task D - Round-Trip Delivery
User daryll
Language Python (PyPy 3.11-v7.3.20)
Score 400
Code Size 833 Byte
Status AC
Exec Time 592 ms
Memory 196340 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 400 / 400
Status
AC × 3
AC × 75
Set Name Test Cases
Sample sample01.txt, sample02.txt, sample03.txt
All sample01.txt, sample02.txt, sample03.txt, in01.txt, in02.txt, in03.txt, in04.txt, in05.txt, in06.txt, in07.txt, in08.txt, in09.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in48.txt, in49.txt, in50.txt, in51.txt, in52.txt, in53.txt, in54.txt, in55.txt, in56.txt, in57.txt, in58.txt, in59.txt, in60.txt, in61.txt, in62.txt, in63.txt, in64.txt, in65.txt, in66.txt, in67.txt, in68.txt, in69.txt, in70.txt, in71.txt, in72.txt
Case Name Status Exec Time Memory
in01.txt AC 54 ms 82116 KiB
in02.txt AC 54 ms 82168 KiB
in03.txt AC 56 ms 82600 KiB
in04.txt AC 80 ms 102476 KiB
in05.txt AC 55 ms 82008 KiB
in06.txt AC 55 ms 82044 KiB
in07.txt AC 57 ms 83292 KiB
in08.txt AC 80 ms 102976 KiB
in09.txt AC 65 ms 95808 KiB
in10.txt AC 582 ms 180724 KiB
in11.txt AC 573 ms 180320 KiB
in12.txt AC 558 ms 180496 KiB
in13.txt AC 585 ms 184648 KiB
in14.txt AC 226 ms 136632 KiB
in15.txt AC 334 ms 149732 KiB
in16.txt AC 228 ms 136184 KiB
in17.txt AC 226 ms 136788 KiB
in18.txt AC 347 ms 153940 KiB
in19.txt AC 368 ms 165704 KiB
in20.txt AC 557 ms 180064 KiB
in21.txt AC 513 ms 176088 KiB
in22.txt AC 55 ms 82036 KiB
in23.txt AC 55 ms 82272 KiB
in24.txt AC 55 ms 82092 KiB
in25.txt AC 55 ms 82256 KiB
in26.txt AC 55 ms 82088 KiB
in27.txt AC 83 ms 105412 KiB
in28.txt AC 291 ms 148748 KiB
in29.txt AC 439 ms 165156 KiB
in30.txt AC 67 ms 95188 KiB
in31.txt AC 204 ms 137028 KiB
in32.txt AC 140 ms 115684 KiB
in33.txt AC 55 ms 82108 KiB
in34.txt AC 56 ms 82912 KiB
in35.txt AC 54 ms 82272 KiB
in36.txt AC 54 ms 82088 KiB
in37.txt AC 54 ms 82256 KiB
in38.txt AC 54 ms 81828 KiB
in39.txt AC 567 ms 180548 KiB
in40.txt AC 445 ms 170236 KiB
in41.txt AC 209 ms 137428 KiB
in42.txt AC 548 ms 180456 KiB
in43.txt AC 553 ms 180432 KiB
in44.txt AC 564 ms 180476 KiB
in45.txt AC 361 ms 165008 KiB
in46.txt AC 220 ms 133196 KiB
in47.txt AC 567 ms 180632 KiB
in48.txt AC 566 ms 180528 KiB
in49.txt AC 519 ms 173964 KiB
in50.txt AC 361 ms 165552 KiB
in51.txt AC 592 ms 182268 KiB
in52.txt AC 458 ms 166136 KiB
in53.txt AC 55 ms 82272 KiB
in54.txt AC 54 ms 82092 KiB
in55.txt AC 55 ms 82468 KiB
in56.txt AC 55 ms 82716 KiB
in57.txt AC 54 ms 81984 KiB
in58.txt AC 55 ms 82272 KiB
in59.txt AC 54 ms 82220 KiB
in60.txt AC 54 ms 81984 KiB
in61.txt AC 55 ms 82144 KiB
in62.txt AC 54 ms 82144 KiB
in63.txt AC 69 ms 96928 KiB
in64.txt AC 217 ms 147692 KiB
in65.txt AC 55 ms 82124 KiB
in66.txt AC 54 ms 82092 KiB
in67.txt AC 54 ms 82044 KiB
in68.txt AC 54 ms 82272 KiB
in69.txt AC 54 ms 82256 KiB
in70.txt AC 351 ms 157120 KiB
in71.txt AC 429 ms 196340 KiB
in72.txt AC 56 ms 83536 KiB
sample01.txt AC 55 ms 81828 KiB
sample02.txt AC 54 ms 82040 KiB
sample03.txt AC 54 ms 82272 KiB