結果

問題 No.277 根掘り葉掘り
コンテスト
ユーザー らっしー(raccy)
提出日時 2017-03-12 12:02:15
言語 Ruby
(3.4.1)
結果
AC  
実行時間 459 ms / 3,000 ms
コード長 702 bytes
コンパイル時間 89 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 31,872 KB
最終ジャッジ日時 2024-06-25 05:18:27
合計ジャッジ時間 7,144 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

# frozen_string_literal: true
n = gets.to_i
edges = Array.new(n + 1) { [] }
(n - 1).times do
  x, y = gets.split.map(&:to_i)
  edges[x] << y
  edges[y] << x
end

zero_nodes = [1] + edges.each_with_index.drop(2)
  .find_all { |v, _i| v.length == 1 }.map(&:last)

node_lengths = [nil] * (n + 1)
checked_nodes = [false] * (n + 1)

queue = []

zero_nodes.each do |z|
  queue << z
  checked_nodes[z] = true
  node_lengths[z] = 0
end

loop do
  i = queue.shift
  next_root_length = node_lengths[i] + 1
  edges[i].each do |t|
    next if checked_nodes[t]
    queue << t
    checked_nodes[t] = true
    node_lengths[t] = next_root_length
  end
  break if queue.empty?
end

node_lengths.shift
puts node_lengths
0