結果

問題 No.1890 Many Sequences Sum Queries
ユーザー 👑 obakyanobakyan
提出日時 2022-04-05 09:18:37
言語 Lua
(LuaJit 2.1.1696795921)
結果
AC  
実行時間 68 ms / 2,000 ms
コード長 856 bytes
コンパイル時間 59 ms
コンパイル使用メモリ 5,248 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-04 19:17:23
合計ジャッジ時間 1,631 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 66 ms
5,248 KB
testcase_01 AC 68 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 3 ms
5,376 KB
testcase_07 AC 31 ms
5,376 KB
testcase_08 AC 9 ms
5,376 KB
testcase_09 AC 11 ms
5,376 KB
testcase_10 AC 59 ms
5,376 KB
testcase_11 AC 27 ms
5,376 KB
testcase_12 AC 59 ms
5,376 KB
testcase_13 AC 57 ms
5,376 KB
testcase_14 AC 32 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 1 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 1 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 1 ms
5,376 KB
testcase_23 AC 1 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

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