結果
問題 | No.1390 Get together |
ユーザー | Kiri8128 |
提出日時 | 2021-02-12 21:26:25 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 464 ms / 2,000 ms |
コード長 | 1,343 bytes |
コンパイル時間 | 232 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 126,464 KB |
最終ジャッジ日時 | 2024-07-19 20:13:55 |
合計ジャッジ時間 | 8,883 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
import sys input = lambda: sys.stdin.readline().rstrip() class UnionFind(): def __init__(self, n): self.n = n self.PA = [-1] * n def root(self, a): L = [] while self.PA[a] >= 0: L.append(a) a = self.PA[a] for l in L: self.PA[l] = a return a def unite(self, a, b): ra, rb = self.root(a), self.root(b) if ra != rb: if self.PA[rb] >= self.PA[ra]: self.PA[ra] += self.PA[rb] self.PA[rb] = ra else: self.PA[rb] += self.PA[ra] self.PA[ra] = rb def size(self, a): return -self.PA[self.root(a)] def groups(self): G = [[] for _ in range(self.n)] for i in range(self.n): G[self.root(i)].append(i) return [g for g in G if g] def group_size(self): G = [[] for _ in range(self.n)] for i in range(self.n): G[self.root(i)].append(i) return [len(g) for g in G if g] N, M = map(int, input().split()) uf = UnionFind(M) C = [[] for _ in range(N)] for _ in range(N): a, b = map(int, input().split()) a, b = a-1, b-1 C[b].append(a) for c in C: if len(c) > 1: i = c[0] for j in c[1:]: uf.unite(i, j) print(M - len(uf.groups()))