結果
問題 |
No.1390 Get together
|
ユーザー |
![]() |
提出日時 | 2021-03-14 19:48:00 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,342 ms / 2,000 ms |
コード長 | 1,137 bytes |
コンパイル時間 | 238 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 75,264 KB |
最終ジャッジ日時 | 2024-11-06 08:52:08 |
合計ジャッジ時間 | 22,603 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
N, M = map(int, input().split()) AB = [tuple(map(int,input().split())) for i in range(N)] graph = [[] for _ in range(M)] class UnionFind: def __init__(self,N): self.parent = [i for i in range(N)] self._size = [1] * N self.count = 0 def root(self,a): if self.parent[a] == a: return a else: self.parent[a] = self.root(self.parent[a]) return self.parent[a] def is_same(self,a,b): return self.root(a) == self.root(b) def unite(self,a,b): ra = self.root(a) rb = self.root(b) if ra == rb: return if self._size[ra] < self._size[rb]: ra,rb = rb,ra self._size[ra] += self._size[rb] self.parent[rb] = ra self.count += 1 def size(self,a): return self._size[self.root(a)] uf = UnionFind(N) for a,b in AB: a,b = a-1,b-1 graph[a].append(b) for g in graph: if len(g) >= 2: for i in range(len(g)-1): uf.unite(g[i], g[i+1]) dp = [-1] * N for g in graph: for gg in g: dp[uf.root(gg)] += 1 break ans = 0 for e in dp: if e >= 1: ans += e print(ans)