結果

問題 No.1890 Many Sequences Sum Queries
ユーザー 👑 obakyan
提出日時 2022-04-05 09:09:50
言語 Lua
(LuaJit 2.1.1734355927)
結果
WA  
実行時間 -
コード長 879 bytes
コンパイル時間 518 ms
コンパイル使用メモリ 5,376 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-26 08:59:50
合計ジャッジ時間 1,874 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 1 WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

local mfl = math.floor

local function comp(a, b)
  return a < b
end

local function lower_bound(ary, x)
  local num = #ary
  if num == 0 then return 1 end
  if not comp(ary[1], x) then return 1 end
  if comp(ary[num], x) then return num + 1 end
  local min, max = 1, num
  while 1 < max - min do
    local mid = mfl((min + max) / 2)
    if comp(ary[mid], x) then
      min = mid
    else
      max = mid
    end
  end
  return max
end

local n, q = io.read("*n", "*n")
local a = {}
for i = 1, n do a[i] = io.read("*n") end
local b = {}
local f = {}
for i = 1, n do
  for j = 1, i do
    table.insert(b, a[j])
    table.insert(f, i)
  end
end
local bsum = {b[1]}
for i = 2, #b do
  bsum[i] = bsum[i - 1] + b[i]
end

for iq = 1, q do
  local z = io.read("*n")
  local lb = lower_bound(bsum, z)
  if #bsum < lb then
    print(-1)
  else
    -- print(f[lb])
    print(lb)
  end
end
0