結果
| 問題 |
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 |
ソースコード
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)
norioc