結果
問題 |
No.556 仁義なきサルたち
|
ユーザー |
👑 ![]() |
提出日時 | 2021-01-29 19:05:35 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 145 ms / 2,000 ms |
コード長 | 2,542 bytes |
コンパイル時間 | 144 ms |
コンパイル使用メモリ | 82,436 KB |
実行使用メモリ | 77,456 KB |
最終ジャッジ日時 | 2024-06-27 05:39:07 |
合計ジャッジ時間 | 2,933 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 22 |
ソースコード
class Union_Find(): def __init__(self,N): """0,1,...,n-1を要素として初期化する. N:要素数 """ self.n=N self.parents=[-1]*N self.rank=[0]*N def find(self, x): """要素xの属している族を調べる. x:要素 """ V=[] while self.parents[x]>=0: V.append(x) x=self.parents[x] for v in V: self.parents[v]=x return x def union(self, x, y): """要素x,yを同一視する. x,y:要素 """ x=self.find(x) y=self.find(y) if x==y: return if self.size(x)>self.size(y): win,lose=x,y elif self.size(x)<self.size(y): win,lose=y,x elif self.find(x)<self.find(y): win,lose=x,y else: win,lose=y,x if self.rank[x]<self.rank[y]: x,y=y,x self.parents[win]+=self.parents[lose] self.parents[lose]=win def size(self, x): """要素xの属している要素の数. x:要素 """ return -self.parents[self.find(x)] def same(self, x, y): """要素x,yは同一視されているか? x,y:要素 """ return self.find(x) == self.find(y) def members(self, x): """要素xが属している族の要素. ※族の要素の個数が欲しいときはsizeを使うこと!! x:要素 """ root = self.find(x) return [i for i in range(self.n) if self.find(i) == root] def roots(self): """族の名前のリスト """ return [i for i, x in enumerate(self.parents) if x < 0] def group_count(self): """族の個数 """ return len(self.roots()) def all_group_members(self): """全ての族の出力 """ X={r:[] for r in self.roots()} for k in range(self.n): X[self.find(k)].append(k) return X def refresh(self): for i in range(self.n): _=self.find(i) def __str__(self): return '\n'.join('{}: {}'.format(r, self.members(r)) for r in self.roots()) def __repr__(self): return self.__str__() #================================================ import sys input=sys.stdin.readline N,M=map(int,input().split()) U=Union_Find(N+1) for _ in range(M): A,B=map(int,input().split()) U.union(A,B) for i in range(1,N+1): print(U.find(i))