結果

問題 No.1890 Many Sequences Sum Queries
ユーザー 👑 obakyanobakyan
提出日時 2022-04-05 09:09:50
言語 Lua
(LuaJit 2.1.1696795921)
結果
WA  
実行時間 -
コード長 879 bytes
コンパイル時間 95 ms
コンパイル使用メモリ 5,376 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-04 19:08:46
合計ジャッジ時間 2,341 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 1 ms
5,376 KB
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
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 = {}
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