結果

問題 No.826 連絡網
ユーザー むらため
提出日時 2019-05-20 05:06:22
言語 Nim
(2.2.0)
結果
AC  
実行時間 65 ms / 2,000 ms
コード長 1,729 bytes
コンパイル時間 2,805 ms
コンパイル使用メモリ 61,680 KB
実行使用メモリ 11,648 KB
最終ジャッジ日時 2024-07-02 01:37:56
合計ジャッジ時間 4,184 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sequtils,math
template times*(n:int,body) = (for _ in 0..<n: body)
template `max=`*(x,y) = x = max(x,y)
template `min=`*(x,y) = x = min(x,y)
proc getchar_unlocked():char {. importc:"getchar_unlocked",header: "<stdio.h>" ,discardable.}
proc scan(): int =
  while true:
    let k = getchar_unlocked()
    if k < '0': return
    result = 10 * result + k.ord - '0'.ord
template useUnionFind() = # 同一集合の判定/マージ が 実質 O(1)
  type UnionFind[T] = ref object
    parent : seq[T]
  proc newUnionFind[T](size:int) : UnionFind[T] =
    new(result)
    result.parent = newSeqUninitialized[T](size)
    for i in 0.int32..<size.int32: result.parent[i] = i
  proc root[T](self:var UnionFind[T],x:T): T =
    if self.parent[x] == x: return x
    self.parent[x] = self.root(self.parent[x])
    return self.parent[x]
  proc same[T](self:var UnionFind[T],x,y:T) : bool = self.root(x) == self.root(y)
  proc merge[T](self:var UnionFind[T],sx,sy:T) : bool {.discardable.} =
    var rx = self.root(sx)
    var ry = self.root(sy)
    if rx == ry : return false
    if self.parent[ry] < self.parent[rx] : swap(rx,ry)
    if self.parent[rx] == self.parent[ry] : self.parent[rx] -= 1
    self.parent[ry] = rx
    return true
  proc count[T](self:var UnionFind[T],x:T):int =
    for p in self.parent:(if p == self.parent[x]: result += 1)
useUnionFind()

let n = scan()
let p = scan()
var F = newUnionFind[int](n+1)
var isPrimes = newSeqWith(n+1,true)
for i in 2..n.float.sqrt.int:
  if not isPrimes[i] : continue
  for j in countup(i,n,i):
    F.merge(i,j)
    if i != j : isPrimes[j] = false
for i in n.float.sqrt.int..n div 2:
  if not isPrimes[i]: continue
  F.merge(2,i)
# echo F.parent
# echo F.parent[p]
echo F.count(p)
0