結果
問題 |
No.556 仁義なきサルたち
|
ユーザー |
![]() |
提出日時 | 2017-09-12 00:02:27 |
言語 | Ruby (3.4.1) |
結果 |
AC
|
実行時間 | 131 ms / 2,000 ms |
コード長 | 1,033 bytes |
コンパイル時間 | 193 ms |
コンパイル使用メモリ | 7,680 KB |
実行使用メモリ | 12,672 KB |
最終ジャッジ日時 | 2024-11-07 15:55:48 |
合計ジャッジ時間 | 3,244 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 22 |
コンパイルメッセージ
Syntax OK
ソースコード
class UnionFind def initialize(n) @rank = Array.new(n, 0) @size = Array.new(n, 1) @parent = [] n.times do |i| @parent[i] = i end end def find(x) if @parent[x] == x x else @parent[x] = find(@parent[x]) end end def unite(x, y) x = find(x) y = find(y) return if x == y if @rank[x] < @rank[y] @parent[y] = x @size[x] += @size[y] else @parent[y] = x @size[x] += @size[y] @rank[x] += 1 if @rank[x] == @rank[y] end end def same(x, y) find(x) == find(y) end def size(x) @size[find(x)] end end N, M = gets.chomp.split.map(&:to_i) uf = UnionFind.new(N) M.times do a, b = gets.chomp.split.map { |i| i.to_i - 1 }.sort next if uf.find(a) == uf.find(b) if uf.size(a) > uf.size(b) uf.unite(a, b) elsif uf.size(a) < uf.size(b) uf.unite(b, a) else if uf.find(a) < uf.find(b) uf.unite(a, b) else uf.unite(b, a) end end end N.times do |i| puts uf.find(i) + 1 end