結果

問題 No.1390 Get together
ユーザー magurofly
提出日時 2021-02-12 21:31:27
言語 Ruby
(3.4.1)
結果
AC  
実行時間 639 ms / 2,000 ms
コード長 716 bytes
コンパイル時間 35 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 32,344 KB
最終ジャッジ日時 2024-07-19 20:26:03
合計ジャッジ時間 12,978 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

class UnionFind
  def initialize(size); @p, @r = [*0...size], [1] * size; end
  def leader(i); j = i; until i == @p[i]; j, i = i, @p[j] = @p[i]; end; i; end
  def merge(i, j); k, l = leader(i), leader(j); return false if k == l; k, l = l, k if @r[k] < @r[l]; @p[l] = k; @r[k] += @r[l]; true; end
  def same?(i, j); leader(i) == leader(j); end
  def size(i); @r[leader(i)]; end
end

N, M = gets.split.map(&:to_i)
slime2box = Array.new(N) { [] }
uf = UnionFind.new(M)
N.times do
    b, c = gets.split.map { |s| s.to_i - 1 }
    slime2box[c] << b
end

count = 0
slime2box.each do |boxes|
    boxes.each_cons(2) do |(i, j)|
        next if uf.same?(i, j)
        uf.merge(i, j)
        count += 1
    end
end

puts count
0