結果
問題 |
No.826 連絡網
|
ユーザー |
|
提出日時 | 2022-12-27 23:25:19 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,004 bytes |
コンパイル時間 | 372 ms |
コンパイル使用メモリ | 82,352 KB |
実行使用メモリ | 95,848 KB |
最終ジャッジ日時 | 2024-11-22 10:51:01 |
合計ジャッジ時間 | 4,634 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | WA * 30 |
ソースコード
import sys input = sys.stdin.readline class UnionFind(): def __init__(self, n): self.par = [-1]*n self.rank = [0]*n self.siz = [1]*n def root(self, x): if self.par[x] == -1: return x self.par[x] = self.root(self.par[x]) return self.par[x] def is_same(self, x, y): return self.root(x) == self.root(y) def unite(self, x, y): if self.is_same(x, y): return False rx = self.root(x) ry = self.root(y) if self.rank[rx] < self.rank[ry]: rx, ry = ry, rx elif self.rank[rx] == self.rank[ry]: self.rank[rx] += 1 self.par[ry] = rx self.siz[rx] += self.siz[ry] return True def size(self, x): return self.siz[self.root(x)] n, p = map(int, input().split()) uf = UnionFind(n+1) for i in range(2, int(n**0.5)+2): for j in range(i, n+1, i): uf.unite(i, j) print(uf.size(p))