結果
問題 | No.2290 UnUnion Find |
ユーザー |
![]() |
提出日時 | 2023-05-05 21:47:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 633 ms / 2,000 ms |
コード長 | 1,599 bytes |
コンパイル時間 | 198 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 134,956 KB |
最終ジャッジ日時 | 2024-11-23 06:49:56 |
合計ジャッジ時間 | 26,377 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 46 |
ソースコード
class UnionFind:def __init__(self, n):self.parent = [-1] * nself.n = nself.cnt = ndef root(self, x):if self.parent[x] < 0:return xelse:self.parent[x] = self.root(self.parent[x])return self.parent[x]def merge(self, x, y):x = self.root(x)y = self.root(y)if x == y:return Falseif self.parent[x] > self.parent[y]:x, y = y, xself.parent[x] += self.parent[y]self.parent[y] = xself.cnt -= 1for yy in groups[y]:groups[x].append(yy)del groups[y]return Truedef same(self, x, y):return self.root(x) == self.root(y)def size(self, x):return -self.parent[self.root(x)]def count(self):return self.cntdef groups(self):res = [[] for _ in range(self.n)]for i in range(self.n):res[self.root(i)].append(i)return [group for group in res if group]n, q = map(int, input().split())queries = [list(map(int, input().split())) for _ in range(q)]uf = UnionFind(n)groups = {}for i in range(n):groups[i] = [i]for query in queries:if query[0] == 1:_, u, v = queryu -= 1v -= 1uf.merge(u, v)else:_, u = queryu -= 1rt = uf.root(u)if uf.count() == 1:print(-1)else:for v in groups:if v != rt:print(groups[v][0] + 1)break