結果
問題 | No.556 仁義なきサルたち |
ユーザー |
![]() |
提出日時 | 2021-03-22 07:28:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 199 ms / 2,000 ms |
コード長 | 1,645 bytes |
コンパイル時間 | 270 ms |
コンパイル使用メモリ | 82,416 KB |
実行使用メモリ | 78,388 KB |
最終ジャッジ日時 | 2024-11-23 16:17:17 |
合計ジャッジ時間 | 2,898 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 22 |
ソースコード
class UnionFind: def __init__(self, n): self.par = [-1]*n self.rank = [0]*n def Find(self, x): if self.par[x] < 0: return x else: self.par[x] = self.Find(self.par[x]) return self.par[x] def Unite(self, x, y): x = self.Find(x) y = self.Find(y) if x != y: if self.rank[x] < self.rank[y]: self.par[y] += self.par[x] self.par[x] = y else: self.par[x] += self.par[y] self.par[y] = x if self.rank[x] == self.rank[y]: self.rank[x] += 1 def Same(self, x, y): return self.Find(x) == self.Find(y) def Size(self, x): return -self.par[self.Find(x)] import sys import io, os input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline n, m = map(int, input().split()) BOSS = list(range(n)) uf = UnionFind(n) for i in range(m): a, b = map(int, input().split()) a, b = a-1, b-1 if uf.Same(a, b): continue if uf.Size(a) > uf.Size(b): boss = BOSS[uf.Find(a)] uf.Unite(a, b) BOSS[uf.Find(a)] = boss elif uf.Size(a) < uf.Size(b): boss = BOSS[uf.Find(b)] uf.Unite(a, b) BOSS[uf.Find(b)] = boss else: if BOSS[uf.Find(a)] < BOSS[uf.Find(b)]: boss = BOSS[uf.Find(a)] uf.Unite(a, b) BOSS[uf.Find(a)] = boss else: boss = BOSS[uf.Find(b)] uf.Unite(a, b) BOSS[uf.Find(b)] = boss for i in range(n): boss = BOSS[uf.Find(i)] print(boss+1)