結果

問題 No.1484 木に数を書き込む問題 / Just Write Numbers! 2
ユーザー 👑 magurofly
提出日時 2021-12-15 02:08:58
言語 Ruby
(4.0.6 + ACL)
コンパイル:
ruby -w -c _filename_
実行:
ruby _filename_
結果
AC  
実行時間 802 ms / 2,000 ms
+ 496µs
コード長 577 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 10 ms
コンパイル使用メモリ 8,576 KB
実行使用メモリ 56,320 KB
最終ジャッジ日時 2026-08-24 07:33:07
合計ジャッジ時間 16,039 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #
raw source code 

N = gets.to_i
graph = Array.new(N) { [] }
sum = 0
(N - 1).times do
  a, b = gets.split.map(&:to_i)
  sum += 1
  graph[a - 1] << [b - 1, 1]
  graph[b - 1] << [a - 1, 1]
end

dfs = ->(start) {
  visited = Array.new(N, false)
  visited[start] = true
  dist = Array.new(N, nil)
  dist[start] = 0
  stack = [start]
  while (u = stack.pop)
    graph[u].each do |v, c|
      next if visited[v]
      visited[v] = true
      dist[v] = dist[u] + c
      stack << v
    end
  end
  dist
}

d1 = dfs[0]
max = d1.max
v = d1.index(max)
d2 = dfs[v]
diameter = d2.max

puts sum * 2 - diameter
0