結果

問題 No.1449 新プロランド
ユーザー hiragn
提出日時 2022-12-14 19:17:25
言語 PyPy3
(7.3.15)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,000 bytes
コンパイル時間 389 ms
コンパイル使用メモリ 82,492 KB
実行使用メモリ 78,112 KB
最終ジャッジ日時 2024-12-24 02:11:15
合計ジャッジ時間 4,261 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import heapify, heappop, heappush
from collections import defaultdict


def main():
    INF = 10 ** 18
    n, m = map(int, input().split())
    g = defaultdict(list)
    for _ in range(m):
        a, b, c = map(int, input().split())
        a -= 1
        b -= 1
        g[a].append([b, c])
        g[b].append([a, c])
    t = list(map(int, input().split()))

    # ダイクストラ
    dist = [[INF] * 501 for _ in range(n)]
    dist[0][0] = 0
    q = [(0, 0, 0)]  # (今の時間, 今いる頂点，おむすびを食べた時間)
    while q:
        cur_t, cur_v, eat_t = heappop(q)
        if dist[cur_v][eat_t] < cur_t:
            continue
        cur_t += t[cur_v]
        eat_t = min(eat_t + t[cur_v], 500)
        for v, cost in g[cur_v]:
            if dist[v][eat_t] > cur_t + cost // eat_t:
                dist[v][eat_t] = cur_t + cost // eat_t
                heappush(q, (cur_t + cost // eat_t, v, eat_t))
    print(min(dist[- 1]))


if __name__ == "__main__":
    main()
0