結果

問題 No.3270 No Coprime Cycles
コンテスト
ユーザー tomerun
提出日時 2025-09-12 23:47:43
言語 Crystal
(1.14.0)
結果
WA  
実行時間 -
コード長 2,468 bytes
コンパイル時間 11,770 ms
コンパイル使用メモリ 312,112 KB
実行使用メモリ 67,932 KB
最終ジャッジ日時 2025-09-12 23:48:20
合計ジャッジ時間 29,987 ms
ジャッジサーバーID
（参考情報） 		judge10 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 14 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

n = read_line.to_i
a = read_line.split.map(&.to_i)
sieve = Array.new(100001, false)
primes = [] of Int64
2.upto(sieve.size - 1) do |i|
  if !sieve[i]
    primes << i
    i.step(to: sieve.size - 1, by: i) do |j|
      sieve[j] = true
    end
  end
end
q = PriorityQueue(Tuple(Int32, Int32, Int32)).new(n)
n.times do |i|
  primes.each do |p|
    if p * p > a[i]
      if a[i] > 1
        q.add({a[i], a[i], i})
      end
      break
    end
    if a[i] % p == 0
      d = p
      a[i] //= p
      while a[i] % p == 0
        d *= p
        a[i] //= p
      end
      q.add({d.to_i, p.to_i, i})
    end
  end
end
uf = UnionFind.new(n)
pos = Array.new(1000001, -1)
ans = 0i64
while q.size > 0
  d, p, i = q.pop
  if pos[p] == -1
    pos[p] = i
  else
    if uf.find(pos[p], i)
      ans += d
    else
      uf.union(pos[p], i)
    end
  end
end
puts ans

class UnionFind
  def initialize(size : Int32)
    @root = Array(Int32).new(size, -1)
  end

  def union(i, j)
    a = root(i)
    b = root(j)
    if a != b
      @root[a] += @root[b]
      @root[b] = a
    end
  end

  def find(i, j)
    root(i) == root(j)
  end

  def root(i)
    return i if @root[i] < 0
    @root[i] = root(@root[i])
  end

  def size(i)
    -@root[root(i)]
  end
end

class PriorityQueue(T)
  def initialize(capacity : Int32)
    @elem = Array(T).new(capacity)
  end

  def initialize(list : Enumerable(T))
    @elem = list.to_a
    1.upto(size - 1) { |i| fixup(i) }
  end

  def size
    @elem.size
  end

  def add(v)
    @elem << v
    fixup(size - 1)
  end

  def top
    @elem[0]
  end

  def pop
    ret = @elem[0]
    last = @elem.pop
    if size > 0
      @elem[0] = last
      fixdown(0)
    end
    ret
  end

  def clear
    @elem.clear
  end

  def decrease_top(new_value : T)
    @elem[0] = new_value
    fixdown(0)
  end

  def to_s(io : IO)
    io << @elem
  end

  private def fixup(index : Int32)
    while index > 0
      parent = (index - 1) // 2
      break if @elem[parent] >= @elem[index]
      @elem[parent], @elem[index] = @elem[index], @elem[parent]
      index = parent
    end
  end

  private def fixdown(index : Int32)
    while true
      left = index * 2 + 1
      break if left >= size
      right = index * 2 + 2
      child = right >= size || @elem[left] > @elem[right] ? left : right
      if @elem[child] > @elem[index]
        @elem[child], @elem[index] = @elem[index], @elem[child]
        index = child
      else
        break
      end
    end
  end
end
0