結果

問題 No.1293 2種類の道路
ユーザー 👑 Kazun
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

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.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 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 __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=0
for t in G:
if t==0:continue
M=set()
X=0
for a in G[t]:
f=V.find(a)
if f in M:
continue
M.add(f)
X+=V.size(a)
K+=X*len(G[t])
print(K-N)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0