結果

問題 No.1484 木に数を書き込む問題 / Just Write Numbers! 2
ユーザー magurofly
提出日時 2021-12-15 02:08:58
言語 Ruby
(3.4.1)
結果
AC  
実行時間 1,237 ms / 2,000 ms
コード長 577 bytes
コンパイル時間 779 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 54,272 KB
最終ジャッジ日時 2024-07-23 19:50:16
合計ジャッジ時間 24,404 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

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