結果

問題 No.277 根掘り葉掘り
コンテスト
ユーザー らっしー(raccy)
提出日時 2017-03-12 12:03:17
言語 Ruby
(3.4.1)
結果
AC  
実行時間 443 ms / 3,000 ms
コード長 794 bytes
コンパイル時間 341 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 32,000 KB
最終ジャッジ日時 2024-06-30 00:20:16
合計ジャッジ時間 7,598 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 = []
next_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]
    next_queue << t
    checked_nodes[t] = true
    node_lengths[t] = next_root_length
  end
  next unless queue.empty?
  break if next_queue.empty?
  queue = next_queue
  next_queue = []
end

node_lengths.shift
puts node_lengths
0