結果

問題 No.2100 [Cherry Alpha C] Two-way Steps
ユーザー lloyz
提出日時 2022-10-14 22:58:19
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,599 bytes
コンパイル時間 470 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 107,120 KB
最終ジャッジ日時 2024-06-26 16:44:56
合計ジャッジ時間 6,593 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10 TLE * 1 -- * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

n, m = map(int, input().split())
H = list(map(int, input().split()))
G = defaultdict(list)
for _ in range(m):
    x, y = map(int, input().split())
    x -= 1
    y -= 1
    G[x].append((y, H[y] - H[x]))
    G[y].append((x, H[x] - H[y]))

DP = [[-1, -1] for _ in range(n)]
DP[0][0] = 0
H = [(0, 0, 0)]
heapify(H)
while H:
    cnt, cp, up = heappop(H)
    cnt *= -1
    if cnt < DP[cp][up]:
        continue
    if cp == n - 1:
        continue
    for np, d in G[cp]:
        if np < cp:
            continue
        if d > 0:
            if up == 0:
                if cnt + d <= DP[np][1]:
                    continue
                DP[np][1] = cnt + d
                heappush(H, (-DP[np][1], np, up + 1))
        else:
            if cnt <= DP[np][0]:
                continue
            DP[np][0] = cnt
            heappush(H, (-DP[np][0], np, 0))
print(max(DP[n - 1]))

DP = [[-1, -1] for _ in range(n)]
DP[n - 1][0] = 0
H = [(0, n - 1, 0)]
heapify(H)
while H:
    cnt, cp, up = heappop(H)
    cnt *= -1
    if cnt < DP[cp][up]:
        continue
    if cp == 0:
        continue
    for np, d in G[cp]:
        if np > cp:
            continue
        if d > 0:
            if up == 0:
                if cnt + d <= DP[np][1]:
                    continue
                DP[np][1] = cnt + d
                heappush(H, (-DP[np][1], np, up + 1))
        else:
            if cnt <= DP[np][0]:
                continue
            DP[np][0] = cnt
            heappush(H, (-DP[np][0], np, 0))
print(max(DP[0]))
0