結果
問題 | No.1390 Get together |
ユーザー |
👑 ![]() |
提出日時 | 2021-02-12 21:49:30 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 5,151 bytes |
コンパイル時間 | 547 ms |
コンパイル使用メモリ | 82,532 KB |
実行使用メモリ | 94,324 KB |
最終ジャッジ日時 | 2024-07-19 21:01:41 |
合計ジャッジ時間 | 7,006 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 RE * 11 |
ソースコード
class Union_Find():def __init__(self,N):"""0,1,...,n-1を要素として初期化する.N:要素数"""self.n=Nself.parents=[-1]*Nself.rank=[0]*Ndef 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]=xreturn xdef union(self, x, y):"""要素x,yを同一視する.x,y:要素"""x=self.find(x)y=self.find(y)if x==y:returnif self.rank[x]<self.rank[y]:x,y=y,xself.parents[x]+=self.parents[y]self.parents[y]=xif self.rank[x]==self.rank[y]:self.rank[x]+=1def 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 Xdef 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__()#=================================================class Coloring_Union_Find():def __init__(self,N,f,e=0):"""0,1,...,n-1を要素として初期化する.N:要素数f:2変数関数の合成e:最初の値"""self.n=Nself.parents=[-1]*Nself.data=[e]*Nself.rank=[0]*Nself.f=fdef 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]=xreturn xdef union(self, x, y):"""要素x,yを同一視する.x,y:要素"""x=self.find(x)y=self.find(y)if x==y:returnself.data[x]=self.data[y]=self.f(self.data[x],self.data[y])if self.rank[x]<self.rank[y]:x,y=y,xself.parents[x]+=self.parents[y]self.parents[y]=xif self.rank[x]==self.rank[y]:self.rank[x]+=1def 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)]=creturndef 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 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 Xdef 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__()#================================================import sysN,M=map(int,input().split())U=Union_Find(N)X=[-1]*(N+1)for _ in range(N):b,c=map(int,input().split())if X[c]==-1:X[c]=b-1else:U.union(b-1,X[c])print(N-U.group_count())