結果

問題 No.1607 Kth Maximum Card
ユーザー tamato
提出日時 2021-07-16 22:15:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,586 ms / 3,500 ms
コード長 1,341 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 82,188 KB
実行使用メモリ 189,264 KB
最終ジャッジ日時 2024-07-06 09:36:09
合計ジャッジ時間 24,706 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9
L = 2*10**5
LL = L+1


def main():
    import sys
    from collections import deque
    input = sys.stdin.buffer.readline

    N, M, K = map(int, input().split())
    ABC = []
    adj = [[] for _ in range(N+1)]
    C = []
    for i in range(M):
        a, b, c = map(int, input().split())
        ABC.append((a, b, c))
        adj[a].append((b, i))
        adj[b].append((a, i))
        C.append(c)

    ok = L
    ng = -1
    mid = (ok + ng) // 2
    C01 = [0] * M
    while ok - ng > 1:
        for i, c in enumerate(C):
            if c > mid:
                C01[i] = 1
            else:
                C01[i] = 0

        que = deque()
        que.append(LL)
        seen = [-1] * (N+1)
        while que:
            vd = que.popleft()
            v, d = divmod(vd, LL)
            if seen[v] != -1:
                continue
            seen[v] = d
            if v == N:
                break
            for u, i in adj[v]:
                c = C01[i]
                if seen[u] == -1:
                    if c:
                        que.append(u * LL + d + 1)
                    else:
                        que.appendleft(u * LL + d)
        if seen[N] < K:
            ok = mid
        else:
            ng = mid
        mid = (ok + ng) // 2
    print(ok)


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