結果
問題 | No.1293 2種類の道路 |
ユーザー |
👑 ![]() |
提出日時 | 2020-11-20 21:58:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 427 ms / 2,000 ms |
コード長 | 2,467 bytes |
コンパイル時間 | 373 ms |
コンパイル使用メモリ | 82,852 KB |
実行使用メモリ | 110,848 KB |
最終ジャッジ日時 | 2024-07-23 12:58:15 |
合計ジャッジ時間 | 7,305 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
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 __str__(self):return '\n'.join('{}: {}'.format(r, self.members(r)) for r in self.roots())#=====--=========================================N,D,W=map(int,input().split())U=Union_Find(N+1)for _ in range(D):a,b=map(int,input().split())U.union(a,b)V=Union_Find(N+1)for j in range(W):c,d=map(int,input().split())V.union(c,d)G=U.all_group_members()K=0for t in G:if t==0:continueM=set()X=0for a in G[t]:f=V.find(a)if f in M:continueM.add(f)X+=V.size(a)K+=X*len(G[t])print(K-N)