結果

問題 No.1607 Kth Maximum Card
コンテスト
ユーザー shotoyoo
提出日時 2021-07-16 22:30:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,210 ms / 3,500 ms
コード長 1,184 bytes
コンパイル時間 260 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 143,696 KB
最終ジャッジ日時 2024-07-06 09:57:27
合計ジャッジ時間 22,183 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))


# グラフの読み込み・重みあり
n,m,k = map(int, input().split())
ns = [[] for _ in range(n)]
mc = 0
for _ in range(m):
    u,v,c = map(int, input().split())
    u -= 1
    v -= 1
    ns[u].append((c,v))
    ns[v].append((c,u))
    mc = max(mc, c)
from collections import deque
def bfs01(start, ns, x):
    n = len(ns)
    dist = [-1]*n
    dist[start] = 0
    q = deque([start])
    while q:
        u = q.popleft()
        if u==n-1:
            break
        d = dist[u]
        for c,v in ns[u]:
            c = int(c>x)
            if c==1:
                if dist[v]==-1:
                    dist[v] = d + 1
                    q.append(v)
            else:
                if dist[v]==-1 or d<dist[v]:
                    dist[v] = d
                    q.appendleft(v)
    return dist[n-1]<=k-1
ng = -1
ok = mc+1
while abs(ok-ng)>1:
    mm = (ok+ng)//2
    if bfs01(0, ns, mm):
        ok = mm
    else:
        ng = mm
ans = ok
print(ans)
0