結果

問題 No.2316 Freight Train
ユーザー simansiman
提出日時 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

ソースコード

diff #
プレゼンテーションモードにする

class UnionFind
def initialize(n)
@size = Array.new(n, 1)
@rank = Array.new(n, 0)
@parent = []
(0..n).each 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[x] = y
@size[y] += @size[x]
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, 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 == -1
uf.unite(v, i)
end
Q.times do
a, b = gets.split.map(&:to_i)
if uf.same?(a, b)
puts 'Yes'
else
puts 'No'
end
end
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0