結果

問題 No.277 根掘り葉掘り
コンテスト
ユーザー らっしー(raccy)
提出日時 2017-03-12 10:39:41
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 704 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 186 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 33,280 KB
最終ジャッジ日時 2024-06-25 05:16:48
合計ジャッジ時間 9,111 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other WA * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #
raw source code 

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

root_node = 1
leaf_nodes = edges.find_all { |_k, v| v.length == 1 }
  .map(&:first) - [root_node]
zero_nodes = [root_node] + leaf_nodes

node_lengths = Array.new(n + 1)
checked_nodes = Array.new(n + 1) { false }
queue = []

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

while (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
end

puts node_lengths
0