結果
| 問題 |
No.556 仁義なきサルたち
|
| ユーザー |
Coki628
|
| 提出日時 | 2020-09-07 02:10:13 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 161 ms / 2,000 ms |
| コード長 | 2,765 bytes |
| コンパイル時間 | 171 ms |
| コンパイル使用メモリ | 82,524 KB |
| 実行使用メモリ | 78,676 KB |
| 最終ジャッジ日時 | 2024-11-29 07:59:43 |
| 合計ジャッジ時間 | 2,658 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| other | AC * 22 |
ソースコード
import sys
def input(): return sys.stdin.readline().strip()
def list2d(a, b, c): return [[c] * b for i in range(a)]
def list3d(a, b, c, d): return [[[d] * c for j in range(b)] for i in range(a)]
def list4d(a, b, c, d, e): return [[[[e] * d for j in range(c)] for j in range(b)] for i in range(a)]
def ceil(x, y=1): return int(-(-x // y))
def INT(): return int(input())
def MAP(): return map(int, input().split())
def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)]
def Yes(): print('Yes')
def No(): print('No')
def YES(): print('YES')
def NO(): print('NO')
sys.setrecursionlimit(10**9)
INF = 10**19
MOD = 10**9 + 7
EPS = 10**-10
class UnionFind:
""" Union-Find木 """
def __init__(self, n):
self.n = n
self.par = [i for i in range(n)]
self.rank = [0] * n
self.size = [1] * n
self.tree = [True] * n
self.grpcnt = n
# グループのボスの番号
self.boss = [i for i in range(n)]
def find(self, x):
""" 根の検索(グループ番号の取得) """
t = []
while self.par[x] != x:
t.append(x)
x = self.par[x]
for i in t:
self.par[i] = x
return self.par[x]
def union(self, x, y):
""" 併合 """
x = self.find(x)
y = self.find(y)
if x == y:
self.tree[x] = False
return
if not self.tree[x] or not self.tree[y]:
self.tree[x] = self.tree[y] = False
# 条件に合わせてボスを交代
if self.size[x] > self.size[y]:
self.boss[y] = self.boss[x]
elif self.size[x] < self.size[y]:
self.boss[x] = self.boss[y]
else:
self.boss[x] = self.boss[y] = min(self.boss[x], self.boss[y])
self.grpcnt -= 1
if self.rank[x] < self.rank[y]:
self.par[x] = y
self.size[y] += self.size[x]
else:
self.par[y] = x
self.size[x] += self.size[y]
if self.rank[x] == self.rank[y]:
self.rank[x] += 1
def is_same(self, x, y):
""" 同じ集合に属するか判定 """
return self.find(x) == self.find(y)
def get_size(self, x=None):
if x is not None:
""" あるノードの属する集合のノード数 """
return self.size[self.find(x)]
else:
""" 集合の数 """
return self.grpcnt
def is_tree(self, x):
""" 木かどうかの判定 """
return self.tree[self.find(x)]
N, M = MAP()
uf = UnionFind(N)
for i in range(M):
a, b = MAP()
a -= 1; b -= 1
uf.union(a, b)
for i in range(N):
ans = uf.boss[uf.find(i)] + 1
print(ans)
Coki628