結果
問題 | No.1390 Get together |
ユーザー |
![]() |
提出日時 | 2021-02-12 22:43:53 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 531 ms / 2,000 ms |
コード長 | 1,420 bytes |
コンパイル時間 | 358 ms |
コンパイル使用メモリ | 81,976 KB |
実行使用メモリ | 118,400 KB |
最終ジャッジ日時 | 2024-07-19 23:40:35 |
合計ジャッジ時間 | 10,439 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
n, m = map(int, input().split())memo = [set() for _ in range(n)]for _ in range(n):b, c = map(int, input().split())b, c = b - 1, c - 1memo[c].add(b)class UnionFind:# あり本実装# rankとrootの配列を1つで賄う方法def __init__(self, n):self.n = nself.par = [-1] * n# 根を求めるdef find(self, x):if self.par[x] < 0:return xelse:self.par[x] = self.find(self.par[x])return self.par[x]# 同じかどうかの判定def is_same(self, x, y):return self.find(x) == self.find(y)# 集合の大きさdef size(self, x):return -self.par[self.find(x)]# 2つの集合の合体# 重みが大きい方に小さい方をつけるようにするdef unite(self, x, y):x = self.find(x)y = self.find(y)if x == y:return Falseif self.par[x] > self.par[y]:x, y = y, xself.par[x] += self.par[y]self.par[y] = xreturn TrueUF = UnionFind(m)ans = 0for i in range(n):temp = list(memo[i])if len(temp) >= 2:base = temp[0]for i in range(len(temp) - 1):now = temp[i + 1]if UF.is_same(base, now):continueelse:UF.unite(base, now)ans += 1print(ans)