結果

問題 No.489 株に挑戦
ユーザー 👑 obakyan
提出日時 2019-05-02 15:11:44
言語 Lua
(LuaJit 2.1.1734355927)
結果
WA  
実行時間 -
コード長 1,664 bytes
コンパイル時間 269 ms
コンパイル使用メモリ 6,952 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-31 13:26:46
合計ジャッジ時間 2,546 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

local ior = io.input()
local n, d, k = ior:read("*n", "*n", "*n")
local t = {}
for i = 1, n do t[i] = ior:read("*n") end
local ret_stt_pos = 1
local ret_val = 0
local mma = math.max
if(d < 50) then
  for i = 1, n - 1 do
    local max = 0
    local lst = math.min(n, i + d)
    for j = i + 1, lst do
      max = mma(max, t[j])
    end
    if(ret_val < max - t[i]) then
      ret_val = max - t[i]
      ret_stt_pos = i
    end
  end
else
  local sqd = math.floor(math.sqrt(d))
  local blockmax = {}
  local blockcount = math.ceil(n / sqd)
  for i = 1, blockcount do blockmax[i] = 0 end
  for i = 1, n do
    local b_idx = math.ceil(i / sqd)
    blockmax[b_idx] = mma(blockmax[b_idx], t[i])
  end
  for i = 1, n - 1 do
    local max = 0
    local lst = math.min(n, i + d)
    local block_start_idx = math.ceil(i / sqd) + 1
    local block_end_idx = math.ceil(lst / sqd) - 1
    local block_start_pos = (block_start_idx - 1) * sqd + 1
    block_start_pos = math.min(n + 1, block_start_pos)
    local block_end_pos = block_end_idx * sqd
    for j = block_start_idx, block_end_idx do
      max = mma(max, blockmax[j])
    end
    for j = i + 1, block_start_pos - 1 do
      max = mma(max, t[j])
    end
    for j = block_end_pos + 1, lst do
      max = mma(max, t[j])
    end
    if(ret_val < max - t[i]) then
      ret_val = max - t[i]
      ret_stt_pos = i
    end
  end
end
print(ret_val * k)
if(0 < ret_val) then
  local max = t[ret_stt_pos + 1]
  local maxj = ret_stt_pos + 1
  local lst = math.min(n, ret_stt_pos + d)
  for j = ret_stt_pos + 1, lst do
    if(max < t[j]) then
      max, maxj = t[j], j
    end
  end
  print(ret_stt_pos - 1 .. " " .. maxj - 1)
end
0