結果

問題 No.826 連絡網
ユーザー むらため
提出日時 2019-05-20 04:52:39
言語 Nim
(2.2.0)
結果
WA  
実行時間 -
コード長 1,536 bytes
コンパイル時間 2,631 ms
コンパイル使用メモリ 61,272 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-07-02 01:36:18
合計ジャッジ時間 4,838 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other WA * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(1, 8) Warning: imported and not used: 'sequtils' [UnusedImport]

ソースコード

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 i in 0..<self.parent.len:
      if self.parent[i] == self.parent[x]: result += 1
useUnionFind()

let n = scan()
let p = scan()
var F = newUnionFind[int](n+1)
for i in 2..n.float.sqrt.int+1:
  for j in countup(i,n,i):
    F.merge(i,j)
echo F.count(p)
0