結果

問題 No.1484 木に数を書き込む問題 / Just Write Numbers! 2
ユーザー 👑 obakyanobakyan
提出日時 2021-04-20 23:28:31
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 369 ms / 2,000 ms
コード長 807 bytes
コンパイル時間 90 ms
コンパイル使用メモリ 6,684 KB
実行使用メモリ 35,328 KB
最終ジャッジ日時 2024-07-04 05:29:50
合計ジャッジ時間 6,819 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff #

local n = io.read("*n")
local edge = {}
for i = 1, n do
  edge[i] = {}
end
for i = 1, n - 1 do
  local a, b = io.read("*n", "*n")
  table.insert(edge[a], b)
  table.insert(edge[b], a)
end
local asked = {}
for i = 1, n do
  asked[i] = false
end
asked[1] = true
local tasks = {1}
for i = 1, n do
  local src = tasks[i]
  for j = 1, #edge[src] do
    local dst = edge[src][j]
    if not asked[dst] then
      table.insert(tasks, dst)
      asked[dst] = true
    end
  end
end
local spos = tasks[n]
local len = {}
for i = 1, n do
  len[i] = -1
end
len[spos] = 0
tasks = {spos}
for i = 1, n do
  local src = tasks[i]
  for j = 1, #edge[src] do
    local dst = edge[src][j]
    if len[dst] < 0 then
      len[dst] = len[src] + 1
      table.insert(tasks, dst)
    end
  end
end
print(2 * (n - 1) - len[tasks[n]])
0