結果

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

ソースコード

diff # 

mod = 1000000007
eps = 10**-9
L = 2*10**5


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)]
    C01 = {}
    for _ in range(M):
        a, b, c = map(int, input().split())
        ABC.append((a, b, c))
        adj[a].append(b)
        adj[b].append(a)
        C01[a * (N + 1) + b] = 0

    ok = L
    ng = -1
    mid = (ok + ng) // 2
    while ok - ng > 1:
        for a, b, c in ABC:
            if c > mid:
                C01[a * (N + 1) + b] = 1
            else:
                C01[a * (N + 1) + b] = 0

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


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