結果
問題 | No.556 仁義なきサルたち |
ユーザー |
|
提出日時 | 2024-08-02 02:51:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 149 ms / 2,000 ms |
コード長 | 1,705 bytes |
コンパイル時間 | 424 ms |
コンパイル使用メモリ | 82,404 KB |
実行使用メモリ | 78,036 KB |
最終ジャッジ日時 | 2024-08-02 02:51:32 |
合計ジャッジ時間 | 3,126 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 22 |
ソースコード
## https://yukicoder.me/problems/no/556class UnionFind:"""UnionFindの基本的な処理を実装したクラス"""def __init__(self, size):self.root = [i for i in range(size)]self.size = [1] * sizedef get_root(self, v):if v == self.root[v]:return velse:old_root = self.root[v]new_root = self.get_root(old_root)self.root[v] = new_rootreturn new_rootdef merge(self, u, v):root_u = self.get_root(u)root_v = self.get_root(v)if root_u == root_v:return Falseif self.size[root_u] > self.size[root_v]:self.size[root_u] += self.size[root_v]self.root[root_v] = root_uself.root[v] = root_uelif self.size[root_u] < self.size[root_v]:self.size[root_v] += self.size[root_u]self.root[root_u] = root_vself.root[u] = root_velse:if root_u < root_v:self.size[root_u] += self.size[root_v]self.root[root_v] = root_uself.root[v] = root_uelse:self.size[root_v] += self.size[root_u]self.root[root_u] = root_vself.root[u] = root_vreturn Truedef main():N, M = map(int, input().split())ab = []for _ in range(M):a, b = map(int, input().split())ab.append((a - 1 ,b - 1))uf = UnionFind(N)for a, b in ab:uf.merge(a, b)for i in range(N):a = uf. get_root(i)print(a + 1)if __name__ == "__main__":main()