結果

問題 No.1607 Kth Maximum Card
ユーザー 👑 rin204rin204
提出日時 2021-07-16 22:36:02
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,212 bytes
コンパイル時間 214 ms
コンパイル使用メモリ 82,012 KB
実行使用メモリ 98,620 KB
最終ジャッジ日時 2024-07-06 10:05:27
合計ジャッジ時間 6,641 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 11 WA * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import deque

n, m, k = map(int, input().split())
edges = [[] for _ in range(n)]
for _ in range(m):
    u, v, c = map(int, input().split())
    u -= 1
    v -= 1
    edges[u].append((v, c))
    
queue = deque()
queue.append((0, 0))
used = [False] * n
used[0] = True
while queue:
    pos, d = queue.popleft()
    if pos == n - 1:
        break
    for npos, _ in edges[pos]:
        if used[npos]:
            continue
        used[npos] = True
        queue.append((npos, d + 1))
        
if d < k:
    print(0)
    exit()
    
def ok(x):
    dist = [10 ** 20] * n
    queue = deque()
    queue.append((0, 0))
    dist[0] = 0
    while queue:
        pos, d = queue.popleft()
        if dist[pos] != d:
            continue
        for npos, c in edges[pos]:
            if c >= x:
                if dist[npos] > d + 1:
                    dist[npos] = d + 1
                    queue.append((npos, d + 1))
            else:
                if dist[npos] > d:
                    dist[npos] = d
                    queue.appendleft((npos, d))
    return dist[npos] < k
l = 0
r = 200200
while r - l > 1:
    mid = (r + l) // 2
    if ok(mid):
        r = mid
    else:
        l = mid
print(l)

0