結果
問題 |
No.2290 UnUnion Find
|
ユーザー |
![]() |
提出日時 | 2023-05-05 21:42:06 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,269 bytes |
コンパイル時間 | 461 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 105,276 KB |
最終ジャッジ日時 | 2024-11-23 06:38:24 |
合計ジャッジ時間 | 89,656 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 8 WA * 21 TLE * 17 |
ソースコード
import sys import numpy as np def main(): input = sys.stdin.readline n, q = map(int, input().split()) uf = np.full(n, -1, np.int64) leaders = set(range(n)) def find(x): nonlocal uf y = x while uf[x] >= 0: x = uf[x] while uf[y] >= 0: z = uf[y] uf[y] = x y = z return x def union(x, y): nonlocal uf, leaders x, y = find(x), find(y) if x == y: return False if uf[x] > uf[y]: x, y = y, x uf[x] += uf[y] uf[y] = x leaders.remove(y) return True ans = [] tmp = None for i in range(q): t, *x = map(int, input().split()) if t == 1: a, b = x union(a - 1, b - 1) else: a = x[0] - 1 if len(leaders) == 1: ans.append(-1) else: b = leaders.pop() if b == a: tmp = b b = leaders.pop() ans.append(b + 1) leaders.add(b) if tmp is not None: leaders.add(tmp) tmp = None print('\n'.join(map(str, ans))) if __name__ == '__main__': main()