結果

問題 No.806 木を道に
ユーザー siman
提出日時 2020-09-24 14:07:00
言語 Ruby
(3.4.1)
結果
AC  
実行時間 416 ms / 2,000 ms
コード長 513 bytes
コンパイル時間 152 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 27,520 KB
最終ジャッジ日時 2024-06-28 04:53:06
合計ジャッジ時間 7,225 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 27
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

N = gets.to_i
E = Hash.new { |h, k| h[k] = [] }

(N - 1).times do
  u, v = gets.split.map(&:to_i)
  E[u] << v
  E[v] << u
end

ans = 0

f =->(v, parent = -1) {
  cnt = E[v].size - 2
  ans += cnt if cnt > 0

  E[v].each do |u|
    next if u == parent

    f[u, v]
  end
}

root = [*0..N].index { |u| E[u].size == 1 }
queue = [[root, -1]]

until queue.empty?
  v, parent = queue.shift
  cnt = E[v].size - 2
  ans += cnt if cnt > 0

  E[v].each do |u|
    next if u == parent
    queue << [u, v]
  end
end

puts ans
0