結果
問題 | No.1293 2種類の道路 |
ユーザー |
![]() |
提出日時 | 2021-09-15 10:08:03 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 453 ms / 2,000 ms |
コード長 | 1,525 bytes |
コンパイル時間 | 717 ms |
コンパイル使用メモリ | 81,468 KB |
実行使用メモリ | 116,792 KB |
最終ジャッジ日時 | 2024-06-28 09:24:52 |
合計ジャッジ時間 | 7,738 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
class UnionFind:def __init__(self, n):self.par = [-1]*nself.rank = [0]*ndef Find(self, x):if self.par[x] < 0:return xelse:self.par[x] = self.Find(self.par[x])return self.par[x]def Unite(self, x, y):x = self.Find(x)y = self.Find(y)if x != y:if self.rank[x] < self.rank[y]:self.par[y] += self.par[x]self.par[x] = yelse:self.par[x] += self.par[y]self.par[y] = xif self.rank[x] == self.rank[y]:self.rank[x] += 1def Same(self, x, y):return self.Find(x) == self.Find(y)def Size(self, x):return -self.par[self.Find(x)]import sysimport io, osinput = io.BytesIO(os.read(0,os.fstat(0).st_size)).readlinefrom collections import defaultdictn, d, w = map(int, input().split())uf1 = UnionFind(n)for i in range(d):a, b = map(int, input().split())a, b = a-1, b-1uf1.Unite(a, b)uf2 = UnionFind(n)for i in range(w):c, d = map(int, input().split())c, d = c-1, d-1uf2.Unite(c, d)X = []for i in range(n):if uf1.par[i] < 0:X.append(i)CX = defaultdict(lambda: 0)usedY = defaultdict(lambda: set())for i in range(n):x = uf1.Find(i)y = uf2.Find(i)if y in usedY[x]:continueCX[x] += uf2.Size(y)usedY[x].add(y)ans = 0for x in X:ans += uf1.Size(x)*(CX[x]-1)print(ans)