結果

問題 No.1607 Kth Maximum Card
ユーザー convexineq
提出日時 2021-07-16 22:33:38
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,341 ms / 3,500 ms
コード長 794 bytes
コンパイル時間 134 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 146,532 KB
最終ジャッジ日時 2024-07-06 10:02:49
合計ジャッジ時間 15,075 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import deque

def check(mid): # mid より大きい辺を使う最小回数
    d = [1000000]*n
    d[0] = 0
    q = deque([0])
    while q:
        v = q.popleft()
        if v==n-1:
            return d[v]
        for u,c in g[v]:
            cost = int(c > mid)
            nd = d[v] + cost
            if nd < d[u]:
                d[u] = nd
                if cost: q.append(u)
                else: q.appendleft(u)
    return 1000000

import sys
readline = sys.stdin.readline
n,m,k = map(int,readline().split())
g = [[] for _ in range(n)]
for _ in range(m):
    u,v,c = map(int,readline().split())
    g[u-1].append((v-1,c))
    g[v-1].append((u-1,c))
ng = -1
ok = 2*10**5+1
while ok-ng>1:
    mid = (ok+ng)//2
    if check(mid) < k: ok = mid
    else: ng = mid
print(ok)
0