結果

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

ソースコード

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 = {}
for i = 1, n do
  for j = 1, a[i] do
    table.insert(b, j)
  end
end
-- print(table.concat(b, " "))
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(lb)
  end
end
0