結果

問題 No.277 根掘り葉掘り
ユーザー らっしー(raccy)らっしー(raccy)
提出日時 2017-03-12 10:39:41
言語 Ruby
(3.3.0)
結果
WA  
実行時間 -
コード長 704 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 11,172 KB
実行使用メモリ 34,500 KB
最終ジャッジ日時 2023-09-07 11:06:06
合計ジャッジ時間 9,365 ms
ジャッジサーバーID
(参考情報)
judge12 / judge15
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
権限があれば一括ダウンロードができます
コンパイルメッセージ
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

puts node_lengths
0