結果

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

ソースコード

diff # 

# 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
node_lengths.shift
puts node_lengths
0