結果

問題 No.1995 CHIKA Road
ユーザー kept1994kept1994
提出日時 2022-10-30 19:32:44
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 706 ms / 2,000 ms
コード長 2,043 bytes
コンパイル時間 146 ms
コンパイル使用メモリ 82,388 KB
実行使用メモリ 160,128 KB
最終ジャッジ日時 2024-07-07 11:32:30
合計ジャッジ時間 12,357 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
53,484 KB
testcase_01 AC 36 ms
53,048 KB
testcase_02 AC 36 ms
53,828 KB
testcase_03 AC 37 ms
52,932 KB
testcase_04 AC 39 ms
54,132 KB
testcase_05 AC 51 ms
61,496 KB
testcase_06 AC 137 ms
80,776 KB
testcase_07 AC 225 ms
94,168 KB
testcase_08 AC 54 ms
64,964 KB
testcase_09 AC 51 ms
62,564 KB
testcase_10 AC 278 ms
88,764 KB
testcase_11 AC 706 ms
160,128 KB
testcase_12 AC 272 ms
124,248 KB
testcase_13 AC 279 ms
106,232 KB
testcase_14 AC 295 ms
107,344 KB
testcase_15 AC 591 ms
148,108 KB
testcase_16 AC 160 ms
84,864 KB
testcase_17 AC 363 ms
110,192 KB
testcase_18 AC 497 ms
123,208 KB
testcase_19 AC 495 ms
123,168 KB
testcase_20 AC 332 ms
109,084 KB
testcase_21 AC 317 ms
106,288 KB
testcase_22 AC 480 ms
121,156 KB
testcase_23 AC 229 ms
93,836 KB
testcase_24 AC 458 ms
123,180 KB
testcase_25 AC 187 ms
86,524 KB
testcase_26 AC 251 ms
96,300 KB
testcase_27 AC 445 ms
121,904 KB
testcase_28 AC 315 ms
106,152 KB
testcase_29 AC 527 ms
124,992 KB
testcase_30 AC 167 ms
86,260 KB
testcase_31 AC 142 ms
79,812 KB
testcase_32 AC 188 ms
90,416 KB
testcase_33 AC 436 ms
121,540 KB
testcase_34 AC 157 ms
80,552 KB
testcase_35 AC 242 ms
94,996 KB
testcase_36 AC 268 ms
98,120 KB
testcase_37 AC 479 ms
123,604 KB
testcase_38 AC 369 ms
108,076 KB
testcase_39 AC 139 ms
79,480 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/usr/bin/env python3
import heapq
INF = 10 ** 16
class Dijkstra():
    def __init__(self, N: int) -> None:
        self.N = N 
        self.G = [[] for _ in range(N)]
        return
    
    # 辺の追加
    def addEdge(self, fromNode: int, toNode: int, cost: int):
        self.G[fromNode].append((cost, toNode))
        return
    
    def build(self, startNode: int):
        hq = []
        heapq.heapify(hq)
        # Set start info
        dist = [INF] * self.N
        # prev = [-1] * self.N # 経路復元する場合は移動時に直前の頂点や辺を記録して遷移していく。
        heapq.heappush(hq, (0, startNode))
        dist[startNode] = 0
        # dijkstra
        while hq:
            min_cost, now = heapq.heappop(hq)
            if min_cost > dist[now]:
                continue
            for cost, next in self.G[now]:
                if dist[next] > dist[now] + cost:
                    dist[next] = dist[now] + cost
                    # prev[next] = now # 頂点nextに至る直前の頂点を更新。
                    heapq.heappush(hq, (dist[next], next))
        return dist

def main():
    N, M = map(int, input().split())
    A, B = [], []
    city = set()
    for i in range(M):
        aa, bb = map(int, input().split())
        A.append(aa)
        B.append(bb)
        city.add(aa)
        city.add(bb)
    compressed = {}
    compressed_to_raw = []
    city.add(1)
    city.add(N)
    city = sorted(list(city))
    for index, val in enumerate(city):
        compressed[val] = index
        compressed_to_raw.append(val)
    dk = Dijkstra(len(city))
    for i in range(len(city) - 1):
        aa = city[i]
        bb = city[i + 1]
        s = compressed[aa]
        t = compressed[bb]
        d = (bb - aa) * 2
        dk.addEdge(s, t, d)
    for aa, bb in zip(A, B):
        s = compressed[aa]
        t = compressed[bb]
        d = 2 * bb - 2 * aa - 1
        dk.addEdge(s, t, d)
    res = dk.build(startNode=0)
    print(res[-1])
    return


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