結果

問題 No.1373 Directed Operations
ユーザー 👑 obakyanobakyan
提出日時 2021-07-19 22:24:44
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 82 ms / 2,000 ms
コード長 662 bytes
コンパイル時間 160 ms
コンパイル使用メモリ 6,684 KB
実行使用メモリ 15,360 KB
最終ジャッジ日時 2024-07-17 13:36:12
合計ジャッジ時間 3,328 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

local n = io.read("*n")
local t = {}
for i = 1, n - 1 do
  t[i] = {}
end
for i = 1, n - 1 do
  local a = io.read("*n")
  table.insert(t[a], i)
end
for i = 1, n - 1 do
  for j = 1, #t[i] do
    local k = #t[i] + 1 - j
    if j < k then
      t[i][j], t[i][k] = t[i][k], t[i][j]
    else break
    end
  end
end
local ret = {}
for i = 1, n - 1 do
  ret[i] = 0
end
local curlen = 1
for dst = 2, n do
  while curlen < n and #t[curlen] == 0 do
    curlen = curlen + 1
  end
  if (dst - 1) < curlen then
    print("NO") os.exit()
  end
  local q = t[curlen][#t[curlen]]
  table.remove(t[curlen])
  ret[q] = dst - curlen
end
print("YES")
print(table.concat(ret, "\n"))
0