結果
問題 | No.1293 2種類の道路 |
ユーザー |
![]() |
提出日時 | 2024-02-02 12:04:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 449 ms / 2,000 ms |
コード長 | 1,510 bytes |
コンパイル時間 | 268 ms |
コンパイル使用メモリ | 82,468 KB |
実行使用メモリ | 106,624 KB |
最終ジャッジ日時 | 2024-09-28 10:34:25 |
合計ジャッジ時間 | 6,618 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
import sysinput = sys.stdin.readlineclass UnionFind(object):def __init__(self, n=1):self.par = [i for i in range(n)]self.rank = [0 for _ in range(n)]self.size = [1 for _ in range(n)]def find(self, x):if self.par[x] == x:return xelse:self.par[x] = self.find(self.par[x])return self.par[x]def union(self, x, y):x = self.find(x)y = self.find(y)if x != y:if self.rank[x] < self.rank[y]:x, y = y, xif self.rank[x] == self.rank[y]:self.rank[x] += 1self.par[y] = xself.size[x] += self.size[y]def is_same(self, x, y):return self.find(x) == self.find(y)def get_size(self, x):x = self.find(x)return self.size[x]N, D, W = map(int, input().split())Ux, Uy = UnionFind(N), UnionFind(N)for i in range(D):a, b = map(int, input().split())a, b = a - 1, b - 1Ux.union(a, b)for i in range(W):a, b = map(int, input().split())a, b = a - 1, b - 1Uy.union(a, b)S = set()mem = [[] for i in range(N)]for i in range(N):mem[Ux.find(i)].append(i)ans = 0for i in range(N):uxi = Ux.find(i)if uxi in S:continueS.add(uxi)temp = set()for u in mem[uxi]:temp.add(Uy.find(u))cnt = -1for u in temp:cnt += Uy.get_size(u)ans += len(mem[uxi]) * cntprint(ans)