結果

問題 No.1995 CHIKA Road
コンテスト
ユーザー lloyz
提出日時 2022-07-01 23:18:37
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,082 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 214,256 KB
最終ジャッジ日時 2024-11-26 07:15:08
合計ジャッジ時間 75,037 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14 TLE * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

n, m = map(int, input().split())
Edge = defaultdict(list)
A = set()
for _ in range(m):
    a, b = map(int, input().split())
    Edge[a - 1].append((b - 1, 2 * b - 2 * a - 1))
    A.add(a - 1)
for c in Edge.keys():
    Edge[c].sort(key=lambda x: x[0])
A.add(n - 1)
sortA = sorted(A)
m = len(sortA)

INF = 10**18
DP = defaultdict(lambda: INF)
DP[0] = 0
H = [0]
seen = set()
heapify(H)
while H:
    cp = heappop(H)
    if cp == n - 1:
        print(DP[n - 1])
        break
    ccost = DP[cp]
    for np, dcost in Edge[cp]:
        if ccost + dcost > DP[np]:
            break
        DP[np] = ccost + dcost
        if np in seen:
            continue
        seen.add(np)
        heappush(H, np)
    idx = bisect_right(sortA, cp)
    for i in range(idx, m):
        np = sortA[i]
        if ccost + 2 * (np - cp) > DP[np]:
            break
        DP[np] = ccost + 2 * (np - cp)
        if np in seen:
            continue
        seen.add(np)
        heappush(H, np)
0