結果
問題 | No.2316 Freight Train |
ユーザー |
![]() |
提出日時 | 2023-05-28 10:13:40 |
言語 | Ruby (3.4.1) |
結果 |
AC
|
実行時間 | 1,058 ms / 2,000 ms |
コード長 | 896 bytes |
コンパイル時間 | 255 ms |
コンパイル使用メモリ | 7,296 KB |
実行使用メモリ | 34,560 KB |
最終ジャッジ日時 | 2024-12-26 17:01:07 |
合計ジャッジ時間 | 20,779 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
コンパイルメッセージ
Syntax OK
ソースコード
class UnionFinddef initialize(n)@size = Array.new(n, 1)@rank = Array.new(n, 0)@parent = [](0..n).each do |i|@parent[i] = iendenddef find(x)if @parent[x] == xxelse@parent[x] = find(@parent[x])endenddef unite(x, y)x = find(x)y = find(y)return if x == yif @rank[x] < @rank[y]@parent[x] = y@size[y] += @size[x]else@parent[y] = x@size[x] += @size[y]@rank[x] += 1 if @rank[x] == @rank[y]endenddef same?(x, y)find(x) == find(y)enddef size(x)@size[find(x)]endendN, Q = gets.split.map(&:to_i)P = gets.split.map(&:to_i)uf = UnionFind.new(N + 1)P.each.with_index(1) do |v, i|next if v == -1uf.unite(v, i)endQ.times doa, b = gets.split.map(&:to_i)if uf.same?(a, b)puts 'Yes'elseputs 'No'endend