結果
問題 | No.2072 Anatomy |
ユーザー | 👑 Kazun |
提出日時 | 2022-09-16 22:29:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 302 ms / 2,000 ms |
コード長 | 3,545 bytes |
コンパイル時間 | 205 ms |
コンパイル使用メモリ | 82,732 KB |
実行使用メモリ | 92,616 KB |
最終ジャッジ日時 | 2024-12-21 21:44:39 |
合計ジャッジ時間 | 6,583 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 27 |
ソースコード
class Coloring_Union_Find(): __slots__=["n","parents","rank","data","merge","__group_number"] def __init__(self,N,merge,unit): """ 0,1,...,N-1 を要素として初期化する. N: 要素数 merge: 合成の方法 e: 最初の値 """ self.n=N self.parents=[-1]*N self.data=[unit]*N self.rank=[0]*N self.merge=merge self.__group_number=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 self.__group_number-=1 self.data[x]=self.data[y]=self.merge(self.data[x],self.data[y]) if self.rank[x]<self.rank[y]: x,y=y,x self.parents[x]+=self.parents[y] self.parents[y]=x if self.rank[x]==self.rank[y]: self.rank[x]+=1 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 set(self,x,c): """ 要素 x の属する族の色を c に変更する. x: 要素 c: 色 """ self.data[self.find(x)]=c def look(self,x): """ 要素 x の属する成分の色 x: 要素 """ return self.data[self.find(x)] 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 self.__group_number 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 color_list(self): return [self.look(x) for x in range(self.n)] def color_map(self): return {x:self.look(x) for x in self.roots()} def __str__(self): return '\n'.join('{} [color: {}]: {}'.format(r,self.look(r),self.members(r)) for r in self.roots()) def __repr__(self): return self.__str__() def __getitem__(self,index): return self.data[self.find(index)] def __setitem__(self,index,value): self.data[self.find(index)]=value #================================================== def solve(): import sys input=sys.stdin.readline write=sys.stdout.write N,M=map(int,input().split()) Edge=[] for _ in range(M): u,v=map(int,input().split()) Edge.append((u,v)) Edge.reverse() U=Coloring_Union_Find(N+1, max, 0) for u,v in Edge: U.union(u,v) U.set(u, U.look(u)+1) return U.look(1) print(solve())