結果
問題 | No.1293 2種類の道路 |
ユーザー |
![]() |
提出日時 | 2020-11-20 22:23:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 526 ms / 2,000 ms |
コード長 | 1,239 bytes |
コンパイル時間 | 289 ms |
コンパイル使用メモリ | 82,152 KB |
実行使用メモリ | 97,012 KB |
最終ジャッジ日時 | 2024-07-23 13:18:41 |
合計ジャッジ時間 | 7,003 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
class UnionFind:def __init__(self, n):self.par = [i for i in range(n+1)]self.rank = [0] * (n + 1)self.size = [1] * (n + 1)# 検索def find(self, x):if self.par[x] == x: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:returnif self.rank[x] < self.rank[y]:self.par[x] = yself.size[y] += self.size[x]else:self.par[y] = xself.size[x] += self.size[y]if self.rank[x] == self.rank[y]:self.rank[x] += 1# 同じ集合に属するか判定def same_check(self, x, y):return self.find(x) == self.find(y)n, d, w = map(int, input().split())ufd = UnionFind(n)ufw = UnionFind(n)for i in range(d):a, b = map(int, input().split())ufd.unite(a - 1, b - 1)for i in range(w):c, d = map(int, input().split())ufw.unite(c - 1, d - 1)s = set()for i in range(n):s.add((ufd.find(i), ufw.find(i)))ans = 0for p, q in s:ans += ufd.size[p] * ufw.size[q]print(ans - n)