結果

問題 No.1607 Kth Maximum Card
ユーザー norioc
提出日時 2025-06-11 02:08:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,747 ms / 3,500 ms
コード長 1,202 bytes
コンパイル時間 659 ms
コンパイル使用メモリ 82,336 KB
実行使用メモリ 282,844 KB
最終ジャッジ日時 2025-06-11 02:08:51
合計ジャッジ時間 30,090 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict, deque


def bsearch(low: int, high: int, fun, is_complement=False) -> int:
    def pred(x: int) -> bool:
        return not fun(x) if is_complement else fun(x)

    lo = low
    hi = high
    res = low
    while lo <= hi:
        m = (lo + hi) // 2
        if pred(m):
            res = max(res, m)
            lo = m + 1
        else:
            hi = m - 1

    return res + 1 if is_complement else res


N, M, K = map(int, input().split())

adj = defaultdict(list)
for _ in range(M):
    u, v, c = map(int, input().split())
    u -= 1
    v -= 1
    adj[u].append((v, c))
    adj[v].append((u, c))


# k 番目にくるカードを m 以下にできるか
def can(m: int) -> bool:
    if m < 0: return False

    q = deque([(0, 0)])
    used = set()
    while q:
        v, step = q.popleft()
        if v == N-1:
            return step < K
        if v in used: continue
        used.add(v)

        for to, c in adj[v]:
            if to in used: continue
            if c > m:
                q.append((to, step+1))
            else:
                q.appendleft((to, step))

    return False


ans = bsearch(-1, 2*10**5, can, is_complement=True)
print(ans)
0